Spaces:

cmu-lti
/

sotopia-space

Running on Zero

wdplx commited on May 1

Commit

1417b59

•

1 Parent(s): 7de04aa

changed decorator back to generate

Files changed (1) hide show

sotopia_generate.py CHANGED Viewed

@@ -82,7 +82,6 @@ def generate_action(
     #     return AgentAction(action_type="none", argument="")
 @cache
-@spaces.GPU(600)
 def prepare_model(model_name):
     compute_type = torch.float16
@@ -150,7 +149,7 @@ def obtain_chain_hf(
     chain = LLMChain(llm=hf, prompt=chat_prompt_template)
     return chain
 def generate(
     model_name: str,
     template: str,

     #     return AgentAction(action_type="none", argument="")
 @cache
 def prepare_model(model_name):
     compute_type = torch.float16
     chain = LLMChain(llm=hf, prompt=chat_prompt_template)
     return chain
+@spaces.GPU
 def generate(
     model_name: str,
     template: str,