Devie commited on
Commit
461e7b1
1 Parent(s): a085f4d

Add application file

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -213,10 +213,10 @@ def QA_model():
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",
220
  "repetition_penalty":1.1}
221
  )
222
  print(model.model_id)
 
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",
220
  "repetition_penalty":1.1}
221
  )
222
  print(model.model_id)