BUAADreamer commited on
Commit
3d1995a
·
verified ·
1 Parent(s): e3a1a6c

modify one line to support full sft and inference of llamafactory

Browse files
Files changed (1) hide show
  1. modeling_minicpmo.py +2 -2
modeling_minicpmo.py CHANGED
@@ -751,7 +751,7 @@ class MiniCPMO(MiniCPMOPreTrainedModel):
751
  input_ids=None,
752
  pixel_values=None,
753
  tgt_sizes=None,
754
- audio_features=None,
755
  audio_feature_lens=None,
756
  image_bound=None,
757
  audio_bounds=None,
@@ -3256,4 +3256,4 @@ def prepare_inputs_for_generation(
3256
  "attention_mask": attention_mask,
3257
  }
3258
  )
3259
- return model_inputs
 
751
  input_ids=None,
752
  pixel_values=None,
753
  tgt_sizes=None,
754
+ audio_features=[],
755
  audio_feature_lens=None,
756
  image_bound=None,
757
  audio_bounds=None,
 
3256
  "attention_mask": attention_mask,
3257
  }
3258
  )
3259
+ return model_inputs