sugiv commited on
Commit
14e48d1
1 Parent(s): d6ab42e

Leetmonkey In Action via Inference

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -33,7 +33,7 @@ def download_model(model_name):
33
  model_path = download_model(MODEL_NAME)
34
  llm = Llama(
35
  model_path=model_path,
36
- n_ctx=512,
37
  n_threads=8,
38
  n_gpu_layers=1, # Use all available GPU layers
39
  verbose=False,
 
33
  model_path = download_model(MODEL_NAME)
34
  llm = Llama(
35
  model_path=model_path,
36
+ n_ctx=1024,
37
  n_threads=8,
38
  n_gpu_layers=1, # Use all available GPU layers
39
  verbose=False,