lxysl commited on
Commit
8301b1c
·
1 Parent(s): d512446

update model.generate

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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,