aya-expanse-8b

Running on Zero

vilarin commited on about 18 hours ago

Commit

efe6ac3

•

1 Parent(s): 8954208

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -46,7 +46,7 @@ model = AutoModelForCausalLM.from_pretrained(
     device_map="auto",
     quantization_config=quantization_config)
-@spaces.GPU(duration=100)
 def stream_chat(
     message: str,
     history: list,

     device_map="auto",
     quantization_config=quantization_config)
+@spaces.GPU()
 def stream_chat(
     message: str,
     history: list,