Caleb Fahlgren commited on
Commit
fb10a3d
·
1 Parent(s): 467c2a7

set gpu layers to 50

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ conn = duckdb.connect()
21
 
22
  llama = llama_cpp.Llama(
23
  model_path="Hermes-2-Pro-Llama-3-8B-Q8_0.gguf",
24
- n_gpu_layers=-1,
25
  chat_format="chatml",
26
  n_ctx=2048,
27
  verbose=False,
 
21
 
22
  llama = llama_cpp.Llama(
23
  model_path="Hermes-2-Pro-Llama-3-8B-Q8_0.gguf",
24
+ n_gpu_layers=50,
25
  chat_format="chatml",
26
  n_ctx=2048,
27
  verbose=False,