runninglsy commited on
Commit
7c18e76
·
1 Parent(s): e0b1758

update modeling_ovis.py

Browse files
Files changed (1) hide show
  1. modeling_ovis.py +1 -0
modeling_ovis.py CHANGED
@@ -504,6 +504,7 @@ class Ovis(OvisPreTrainedModel):
504
  f' but got {type(text_or_conversations)}')
505
 
506
  if frame_selector is not None:
 
507
  conversations, images = frame_selector(conversations=conversations, frames=images, **frame_selector_kwargs)
508
 
509
  # format conversations
 
504
  f' but got {type(text_or_conversations)}')
505
 
506
  if frame_selector is not None:
507
+ frame_selector_kwargs = frame_selector_kwargs or {}
508
  conversations, images = frame_selector(conversations=conversations, frames=images, **frame_selector_kwargs)
509
 
510
  # format conversations