Spaces:
Runtime error
Runtime error
extend GPU duration
Browse files- sotopia_generate.py +1 -1
sotopia_generate.py
CHANGED
|
@@ -149,7 +149,7 @@ def obtain_chain_hf(
|
|
| 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,
|
|
|
|
| 149 |
chain = LLMChain(llm=hf, prompt=chat_prompt_template)
|
| 150 |
return chain
|
| 151 |
|
| 152 |
+
@spaces.GPU(duration=120)
|
| 153 |
def generate(
|
| 154 |
model_name: str,
|
| 155 |
template: str,
|