Packages: Fix exllamav2 version check

Post versions are ok to use for checking if the user is on the correct exllamav2 wheel. Signed-off-by: kingbri <[email protected]>
theroyallab · Feb 10, 2024 · 9f1d891 · 9f1d891
1 parent 8d8cf5d
commit 9f1d891
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/backends/exllamav2/utils.py b/backends/exllamav2/utils.py
@@ -9,10 +9,10 @@
 def check_exllama_version():
     """Verifies the exllama version"""
 
-    required_version = "0.0.13"
-    current_version = package_version("exllamav2").split("+")[0]
+    required_version = version.parse("0.0.13.post1")
+    current_version = version.parse(package_version("exllamav2").split("+")[0])
 
-    if version.parse(current_version) < version.parse(required_version):
+    if current_version < required_version:
         raise SystemExit(
             f"ERROR: TabbyAPI requires ExLlamaV2 {required_version} "
             f"or greater. Your current version is {current_version}.\n"