whoami02 commited on
Commit
ade8c24
·
verified ·
1 Parent(s): 2701ca2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -37,9 +37,9 @@ def load_quantized_model(model_id=None):
37
  )
38
  kwargs = {
39
  'model_path': model_path,
40
- 'n_ctx': 10000,
41
- 'max_tokens': 10000,
42
- 'n_batch': 512,
43
  # 'n_gpu_layers':6,
44
  }
45
  return llamacpp.LlamaCpp(**kwargs)
 
37
  )
38
  kwargs = {
39
  'model_path': model_path,
40
+ 'n_ctx': 20000,
41
+ 'max_tokens': 15000,
42
+ 'n_batch': 1024,
43
  # 'n_gpu_layers':6,
44
  }
45
  return llamacpp.LlamaCpp(**kwargs)