Spaces:

cmu-lti
/

sotopia-space

Runtime error

wdplx commited on May 1

Commit

cf47cc0

•

1 Parent(s): 79002f9

extend GPU duration

Files changed (1) hide show

sotopia_generate.py CHANGED Viewed

@@ -149,7 +149,7 @@ def obtain_chain_hf(
     chain = LLMChain(llm=hf, prompt=chat_prompt_template)
     return chain
-@spaces.GPU
 def generate(
     model_name: str,
     template: str,

     chain = LLMChain(llm=hf, prompt=chat_prompt_template)
     return chain
+@spaces.GPU(duration=120)
 def generate(
     model_name: str,
     template: str,