Devie commited on
Commit
a085f4d
1 Parent(s): ffe08cf

Add application file

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -213,7 +213,7 @@ def QA_model():
213
  task="text-generation",
214
  model_kwargs={
215
  "torch_dtype" : torch.float32,
216
- "low_cpu_mem_usage" : True,
217
  "temperature": 0.2,
218
  "max_length": 2048,
219
  #"device_map": "auto",
 
213
  task="text-generation",
214
  model_kwargs={
215
  "torch_dtype" : torch.float32,
216
+ "low_cpu_mem_usage" : False,
217
  "temperature": 0.2,
218
  "max_length": 2048,
219
  #"device_map": "auto",