Update packages.py

xin-li-67 · Jul 7, 2024 · f84b007 · f84b007
1 parent 322663b
commit f84b007
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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")
+    return _get_package_version("vllm") >= version.parse("0.5.1")