BUAADreamer
commited on
modify one line to support full sft and inference of llamafactory
Browse files- 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=
|
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
|