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 |
|