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")