423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-11-05 18:32:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update packages.py

Browse Source 
Former-commit-id: c61ee780f3aed51c31a81e912f25fbfd11dc7edd
This commit is contained in:
hoshi-hiyouga 2024-07-07 15:48:29 +08:00 committed by GitHub
parent 25e086e02d
commit 2e11c6ecdc
1 changed files with 2 additions and 1 deletions
Show Stats 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():
return _get_package_version("vllm") >= version.parse("0.5.0")
@lru_cache
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")