Spaces:

cafe3310
/

ling-playground-basic

Running on Zero

cafe3310 commited on Sep 24

Commit

ce62afe

1 Parent(s): feb5d28

fix: Increase GPU duration to 150s

Files changed (1) hide show

comp.py CHANGED Viewed

@@ -14,7 +14,7 @@ model = AutoModelForCausalLM.from_pretrained(
     trust_remote_code=True
 )
-@spaces.GPU(duration=120)
 def generate_response(message, history, system_prompt=None):
     # (msg, history, system_prompt) -> str: stream response (yielding partial responses)

     trust_remote_code=True
 )
+@spaces.GPU(duration=150)
 def generate_response(message, history, system_prompt=None):
     # (msg, history, system_prompt) -> str: stream response (yielding partial responses)