423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-12-14 19:06:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update packages.py

Browse Source
This commit is contained in:
hoshi-hiyouga
2024-07-07 15:48:29 +08:00
committed by GitHub
parent 322663bf90
commit f84b007ebb
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -82,6 +82,7 @@ def is_vllm_available():
def is_vllm_version_greater_than_0_5(): def is_vllm_version_greater_than_0_5():
return _get_package_version("vllm") >= version.parse("0.5.0") return _get_package_version("vllm") >= version.parse("0.5.0")
@lru_cache @lru_cache
def is_vllm_version_greater_than_0_5_1(): def is_vllm_version_greater_than_0_5_1():
return _get_package_version("vllm") >= version.parse("0.5.1") return _get_package_version("vllm") >= version.parse("0.5.1")