From f84b007ebbb9fa63f797b4bd1c487372877bbc65 Mon Sep 17 00:00:00 2001 From: hoshi-hiyouga Date: Sun, 7 Jul 2024 15:48:29 +0800 Subject: [PATCH] Update packages.py --- 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 1ba4cfbb1a..a9072103a2 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")