required transformers version

Former-commit-id: d9915db327a038c93b5e3421c90b1f218fb23f92
2024-10-14 21:11:09 +08:00
parent 9d929897ce
commit 2b3b0473cd
2 changed files with 7 additions and 1 deletions
--- a/src/llamafactory/data/mm_plugin.py
+++ b/src/llamafactory/data/mm_plugin.py
@@ -692,6 +692,12 @@ def get_mm_plugin(
    video_token: Optional[str] = None,
 ) -> "BasePlugin":
    plugin_class = PLUGINS.get(name, None)
+    if plugin_class == "PixtralPlugin":
+        from transformers.utils.versions import require_version
+        try:
+            require_version("transformers==4.46.0.dev0")
+        except Exception as e:
+            raise ImportError("PixtralPlugin requires transformers==4.46.0.dev0. Please install it first.")
    if plugin_class is None:
        raise ValueError("Multimodal plugin `{}` not found.".format(name))