Spaces:
Runtime error
Runtime error
Add application file
Browse files
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 |
-
|
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)
|