Hugging Face's logo

Spaces:
cocktailpeanut
/
SuperPrompt-v1
Running

App Files Community
Nick088 commited on
Commit
fb4ba64
·
verified ·
1 Parent(s): 17cadbf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ else:
12
 
13
  tokenizer = T5Tokenizer.from_pretrained("google/flan-t5-small")
14
 
15
- model = T5ForConditionalGeneration.from_pretrained("roborovski/superprompt-v1", device_map="auto", torch_dtype=auto)
16
  model.to(device)
17
 
18
  def generate(prompt, model_precision_type, max_new_tokens, repetition_penalty, temperature, top_p, top_k, seed):
 
12
 
13
  tokenizer = T5Tokenizer.from_pretrained("google/flan-t5-small")
14
 
15
+ model = T5ForConditionalGeneration.from_pretrained("roborovski/superprompt-v1", device_map="auto", torch_dtype="auto")
16
  model.to(device)
17
 
18
  def generate(prompt, model_precision_type, max_new_tokens, repetition_penalty, temperature, top_p, top_k, seed):