TaruniSwathi commited on
Commit
8da6b91
·
verified ·
1 Parent(s): cdde206

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -32,7 +32,7 @@ llm = Llama(
32
  n_threads=max(1, os.cpu_count() or 2),
33
  n_threads_batch=max(1, os.cpu_count() or 2),
34
  n_batch=128,
35
- n_gpu_layers=0,
36
  verbose=False,
37
  )
38
 
 
32
  n_threads=max(1, os.cpu_count() or 2),
33
  n_threads_batch=max(1, os.cpu_count() or 2),
34
  n_batch=128,
35
+ n_gpu_layers=1,
36
  verbose=False,
37
  )
38