Nick088 commited on
Commit
9f87e56
·
verified ·
1 Parent(s): 955b4f7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -31,7 +31,7 @@ def generate(
31
  seed
32
  ):
33
 
34
- model = T5ForConditionalGeneration.from_pretrained("roborovski/superprompt-v1", torch_dtype=f"{precision_model}")
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