diff --git a/src/llamafactory/data/mm_plugin.py b/src/llamafactory/data/mm_plugin.py index 92db669eb..5cf0cca8a 100644 --- a/src/llamafactory/data/mm_plugin.py +++ b/src/llamafactory/data/mm_plugin.py @@ -1438,6 +1438,8 @@ class MiniCPMVPlugin(BasePlugin): mm_inputs.update(audio_inputs) mm_inputs.update({"audio_bounds": audio_bounds_ls, "spk_bounds": spk_bounds_ls}) + return mm_inputs + @dataclass class MiniCPMV4_6Plugin(BasePlugin):