robert commited on
Commit
3b1ba3c
·
1 Parent(s): 90af0e7

Increasing timeout to check shared loading

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -52,7 +52,7 @@ class StopOnSequence(StoppingCriteria):
52
  )
53
 
54
 
55
- @spaces.GPU(duration=120)
56
  def spaces_model_predict(message: str, history: list[tuple[str, str]]):
57
  history_transformer_format = history + [[message, ""]]
58
  stop = StopOnSequence("<|human|>", tokenizer)
 
52
  )
53
 
54
 
55
+ @spaces.GPU(duration=512)
56
  def spaces_model_predict(message: str, history: list[tuple[str, str]]):
57
  history_transformer_format = history + [[message, ""]]
58
  stop = StopOnSequence("<|human|>", tokenizer)