Hugging Face's logo

Spaces:
NeoPy
/
SuperPrompt-v1
Runtime error

App Files Community
Nick088 commited on
Commit
7b60be9
·
verified ·
1 Parent(s): 64d6c72

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ else:
13
  print("Using CPU")
14
 
15
  tokenizer = T5Tokenizer.from_pretrained("roborovski/superprompt-v1")
16
- model = T5ForConditionalGeneration.from_pretrained("roborovski/superprompt-v1")
17
  model.to(device)
18
 
19
  @spaces.GPU()
 
13
  print("Using CPU")
14
 
15
  tokenizer = T5Tokenizer.from_pretrained("roborovski/superprompt-v1")
16
+ model = T5ForConditionalGeneration.from_pretrained("roborovski/superprompt-v1", torch_dtype=torch.float16)
17
  model.to(device)
18
 
19
  @spaces.GPU()