abetlen / abetlen/llama-cpp-python

How to use GPU?

Open
#576 21 comments 0 reactions 0 assignees View on GitHub
build hardware
Dominant language
Python
Stars
10.6k
Forks
1.4k
PR merge metrics
PR metrics pending

Description

I run llama cpp python on my new PC which has a built in RTX 3060 with 12GB VRAM
This is my code:
```
from llama_cpp import Llama
llm = Llama(model_path="./wizard-mega-13B.ggmlv3.q4_0.bin", n_ctx=2048)
def generate(params):
print(params["promt"])
output = llm(params["promt"], max_tokens=params["max_tokens"], stop=params["stop"], echo=params["echo"])
```

This code works and I get the results that I want but the inference is terribly slow. for a few tokens it takes up to 10 seconds. How do I minimize this time? I dont think my GPU is doing the heavy lifting here...

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.