[v1] add models & accelerator (#9579)

2025-12-17 04:10:36 +08:00 · 2025-12-08 02:30:25 +08:00
parent 739954910a
commit 5744f1ea94
27 changed files with 335 additions and 105 deletions
--- a/src/llamafactory/data/mm_plugin.py
+++ b/src/llamafactory/data/mm_plugin.py
@@ -465,6 +465,7 @@ class BasePlugin(MMPluginMixin):
        self._validate_input(processor, images, videos, audios)
        return self._get_mm_inputs(images, videos, audios, processor)

+
@dataclass
 class ErnieVLPlugin(BasePlugin):
    @override