Wizz13150 commited on
Commit
7231961
·
1 Parent(s): d9f4047

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -23,9 +23,9 @@ def load_model(model_name):
23
  raise FileNotFoundError(f"Model file not found at {model_path}")
24
  llm = Llama(
25
  model_path=model_path,
26
- n_ctx=2048,
27
  n_threads=2,
28
- n_batch=32,
29
  verbose=False
30
  )
31
  current_model = model_name
 
23
  raise FileNotFoundError(f"Model file not found at {model_path}")
24
  llm = Llama(
25
  model_path=model_path,
26
+ n_ctx=1024,
27
  n_threads=2,
28
+ n_batch=256,
29
  verbose=False
30
  )
31
  current_model = model_name