leoskyscape commited on
Commit
4c2740f
·
verified ·
1 Parent(s): c3d6a36

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -23,8 +23,10 @@ print(f"Model downloaded to: {model_path}")
23
 
24
  llm = Llama(
25
  model_path=model_path,
 
 
26
  n_gpu_layers=0, # GPU acceleration if available
27
- n_ctx=4096,
28
  chat_format="llama-3",
29
  )
30
 
 
23
 
24
  llm = Llama(
25
  model_path=model_path,
26
+ n_threads=2,
27
+ n_batch=256,
28
  n_gpu_layers=0, # GPU acceleration if available
29
+ n_ctx=2048,
30
  chat_format="llama-3",
31
  )
32