granite-3.0-1b-a400m-instruct-CPU

Running

aixsatoshi commited on Jul 23

Commit

d65d6d9

•

1 Parent(s): 9e9c8af

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -40,7 +40,7 @@ h3 {
 }
 """
-@gr.GPU
 def stream_chat(message: str, history: list, temperature: float, max_new_tokens: int, top_p: float, top_k: int, penalty: float):
     print(f'Message: {message}')
     print(f'History: {history}')

 }
 """
+@spaces.GPU(duration=120)
 def stream_chat(message: str, history: list, temperature: float, max_new_tokens: int, top_p: float, top_k: int, penalty: float):
     print(f'Message: {message}')
     print(f'History: {history}')