Syed Junaid Iqbal commited on
Commit
f5f1790
1 Parent(s): 54be36f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -28,7 +28,7 @@ def init_retriever():
28
  template = 0.4,
29
  n_ctx=4000,
30
  max_tokens=4000,
31
- n_gpu_layers = 10,
32
  n_batch = 512,
33
  callback_manager=callback_manager,
34
  verbose=True)
 
28
  template = 0.4,
29
  n_ctx=4000,
30
  max_tokens=4000,
31
+ n_gpu_layers = 50,
32
  n_batch = 512,
33
  callback_manager=callback_manager,
34
  verbose=True)