Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -31,7 +31,7 @@ def generate(
|
|
| 31 |
seed
|
| 32 |
):
|
| 33 |
|
| 34 |
-
model = T5ForConditionalGeneration.from_pretrained("roborovski/superprompt-v1", torch_dtype=
|
| 35 |
|
| 36 |
model.to(device)
|
| 37 |
|
|
|
|
| 31 |
seed
|
| 32 |
):
|
| 33 |
|
| 34 |
+
model = T5ForConditionalGeneration.from_pretrained("roborovski/superprompt-v1", torch_dtype=precision_model)
|
| 35 |
|
| 36 |
model.to(device)
|
| 37 |
|