423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-08-22 22:02:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update packages.py

Browse Source 
Former-commit-id: f84b007ebbb9fa63f797b4bd1c487372877bbc65
This commit is contained in:
hoshi-hiyouga 2024-07-07 15:48:29 +08:00 committed by GitHub
parent 7ca84e0a09
commit 7483e187c6
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")