fix #5747

Former-commit-id: 26d07de349c98b547cd6a6166ea20616d08ba343
2024-10-29 10:47:04 +00:00
parent 248d5daaff
commit e2748fa967
8 changed files with 58 additions and 6 deletions
--- a/src/llamafactory/extras/packages.py
+++ b/src/llamafactory/extras/packages.py
@@ -79,6 +79,11 @@ def is_transformers_version_greater_than_4_43():
    return _get_package_version("transformers") >= version.parse("4.43.0")


+@lru_cache
+def is_transformers_version_equal_to_4_46():
+    return _get_package_version("transformers") == version.parse("4.46.0")
+
+
 def is_uvicorn_available():
    return _is_package_available("uvicorn")