sitammeur commited on
Commit
f2cf41c
·
verified ·
1 Parent(s): 82b8cf4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -70,9 +70,9 @@ def respond(
70
  if llm is None or llm_model != model:
71
  llm = Llama(
72
  model_path=f"models/{model}",
73
- flash_attn=True,
74
- n_gpu_layers=81,
75
- n_batch=1024,
76
  n_ctx=8192,
77
  )
78
  llm_model = model
 
70
  if llm is None or llm_model != model:
71
  llm = Llama(
72
  model_path=f"models/{model}",
73
+ flash_attn=False,
74
+ n_gpu_layers=0,
75
+ n_batch=32,
76
  n_ctx=8192,
77
  )
78
  llm_model = model