423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-12-15 03:10:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix #5747

Browse Source 
Former-commit-id: ae045c884f
This commit is contained in:
hiyouga
2024-10-29 10:47:04 +00:00
parent 0d8aa6e6ef
commit 825ea1c72d
8 changed files with 58 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/llamafactory/extras/packages.py
View File

@@ -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")