Spaces:
Runtime error
Runtime error
Add application file
Browse files
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" :
|
| 217 |
"temperature": 0.2,
|
| 218 |
"max_length": 2048,
|
| 219 |
-
|
| 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)
|