Hugging Face's logo

Spaces:
Mihaiii
/
backtrack_sampler_demo
Sleeping

App Files Community
Mihaiii commited on
Commit
17af049
·
verified ·
1 Parent(s): e878de4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -35,7 +35,7 @@ def create_chat_template_messages(history, prompt):
35
 
36
  return messages
37
 
38
- @spaces.GPU
39
  def generate_responses(prompt, history):
40
  messages = create_chat_template_messages(history, prompt)
41
  wrapped_prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
 
35
 
36
  return messages
37
 
38
+ @spaces.GPU(duration=60)
39
  def generate_responses(prompt, history):
40
  messages = create_chat_template_messages(history, prompt)
41
  wrapped_prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)