Update packages.py

Former-commit-id: c61ee780f3aed51c31a81e912f25fbfd11dc7edd
2024-07-07 15:48:29 +08:00
parent 281fd5bb89
commit 00b93d8b2f
1 changed files with 2 additions and 1 deletions
--- 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")