muryshev commited on
Commit
aa9cfee
1 Parent(s): c15aead

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -133,6 +133,7 @@ def generate_search_request():
133
  logits_all=True,
134
  #n_threads=12,
135
  verbose=True,
 
136
  n_gqa=8 #must be set for 70b models
137
  )
138
 
@@ -182,6 +183,7 @@ def generate_response():
182
  logits_all=True,
183
  #n_threads=12,
184
  verbose=True,
 
185
  n_gqa=8 #must be set for 70b models
186
  )
187
 
 
133
  logits_all=True,
134
  #n_threads=12,
135
  verbose=True,
136
+ n_gpu_layers=20,
137
  n_gqa=8 #must be set for 70b models
138
  )
139
 
 
183
  logits_all=True,
184
  #n_threads=12,
185
  verbose=True,
186
+ n_gpu_layers=20,
187
  n_gqa=8 #must be set for 70b models
188
  )
189