llama-cpp-python How to use vinamrag-code/llama-cpp-python-cuda-wheels with llama-cpp-python:
# !pip install llama-cpp-python
from llama_cpp import Llama
llm = Llama.from_pretrained(
repo_id="vinamrag-code/llama-cpp-python-cuda-wheels",
filename="{{GGUF_FILE}}",
)
output = llm(
"Once upon a time,",
max_tokens=512,
echo=True
)
print(output)
