Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -227,6 +227,7 @@ def gradio_ask(user_message, chatbot):
|
|
227 |
|
228 |
@spaces.GPU
|
229 |
def gradio_answer(chatbot, sys_prompt, user_prompt, video_tensor, chat_state, num_beams, temperature, do_sample=False):
|
|
|
230 |
response, chat_state = model.chat(tokenizer,
|
231 |
sys_prompt,
|
232 |
user_prompt,
|
|
|
227 |
|
228 |
@spaces.GPU
|
229 |
def gradio_answer(chatbot, sys_prompt, user_prompt, video_tensor, chat_state, num_beams, temperature, do_sample=False):
|
230 |
+
video_tensor = video_tensor.to(model.device)
|
231 |
response, chat_state = model.chat(tokenizer,
|
232 |
sys_prompt,
|
233 |
user_prompt,
|