Spaces:
Runtime error
Runtime error
changed decorator back to generate
Browse files- sotopia_generate.py +1 -2
sotopia_generate.py
CHANGED
|
@@ -82,7 +82,6 @@ def generate_action(
|
|
| 82 |
# return AgentAction(action_type="none", argument="")
|
| 83 |
|
| 84 |
@cache
|
| 85 |
-
@spaces.GPU(600)
|
| 86 |
def prepare_model(model_name):
|
| 87 |
compute_type = torch.float16
|
| 88 |
|
|
@@ -150,7 +149,7 @@ def obtain_chain_hf(
|
|
| 150 |
chain = LLMChain(llm=hf, prompt=chat_prompt_template)
|
| 151 |
return chain
|
| 152 |
|
| 153 |
-
|
| 154 |
def generate(
|
| 155 |
model_name: str,
|
| 156 |
template: str,
|
|
|
|
| 82 |
# return AgentAction(action_type="none", argument="")
|
| 83 |
|
| 84 |
@cache
|
|
|
|
| 85 |
def prepare_model(model_name):
|
| 86 |
compute_type = torch.float16
|
| 87 |
|
|
|
|
| 149 |
chain = LLMChain(llm=hf, prompt=chat_prompt_template)
|
| 150 |
return chain
|
| 151 |
|
| 152 |
+
@spaces.GPU
|
| 153 |
def generate(
|
| 154 |
model_name: str,
|
| 155 |
template: str,
|