Yi-6B-Chat

Running on Zero

0x7o commited on Feb 15

Commit

24704b4

•

1 Parent(s): 7093f52

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -22,7 +22,7 @@ class StopOnTokens(StoppingCriteria):
 # Function to generate model predictions.
-@spaces.GPU(duration=180)
 def predict(message, history):
     stop = StopOnTokens()
     conversation = []

 # Function to generate model predictions.
+@spaces.GPU(duration=30)
 def predict(message, history):
     stop = StopOnTokens()
     conversation = []