Spaces:
Running
on
Zero
Running
on
Zero
update model.generate
Browse files
app.py
CHANGED
@@ -340,8 +340,8 @@ def predict(_chatbot, task_history):
|
|
340 |
with torch.inference_mode():
|
341 |
output_ids = model.generate(
|
342 |
input_ids,
|
343 |
-
images=all_visual_tensor,
|
344 |
-
audios=audio_list,
|
345 |
do_sample=False,
|
346 |
temperature=0.01,
|
347 |
top_p=None,
|
|
|
340 |
with torch.inference_mode():
|
341 |
output_ids = model.generate(
|
342 |
input_ids,
|
343 |
+
images=all_visual_tensor if all_visual_tensor else None,
|
344 |
+
audios=audio_list if audio_list else None,
|
345 |
do_sample=False,
|
346 |
temperature=0.01,
|
347 |
top_p=None,
|