plugin test & check

Former-commit-id: 76c7c8c5a729b8b43e3a31efc44f2c9c2678bf3d
2024-10-15 12:12:46 +08:00
parent 2b3b0473cd
commit 3693d7b571
2 changed files with 16 additions and 2 deletions
--- a/src/llamafactory/data/mm_plugin.py
+++ b/src/llamafactory/data/mm_plugin.py
@@ -697,7 +697,7 @@ def get_mm_plugin(
        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.")
+            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))