lvying commited on
Commit
a3a22a7
1 Parent(s): 461e7b1

Add application file

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -216,7 +216,7 @@ def QA_model():
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)
 
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)