modify one line to support full sft and inference of llamafactory
#5
by
BUAADreamer
- opened
- 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
|