From 00b93d8b2f0fca721ad372ca3b881c403c1bf28b Mon Sep 17 00:00:00 2001 From: hoshi-hiyouga Date: Sun, 7 Jul 2024 15:48:29 +0800 Subject: [PATCH] Update packages.py Former-commit-id: c61ee780f3aed51c31a81e912f25fbfd11dc7edd --- src/llamafactory/extras/packages.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/llamafactory/extras/packages.py b/src/llamafactory/extras/packages.py index 1ba4cfbb..a9072103 100644 --- a/src/llamafactory/extras/packages.py +++ b/src/llamafactory/extras/packages.py @@ -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") \ No newline at end of file + return _get_package_version("vllm") >= version.parse("0.5.1")