Only try to install one of the EXLv2 wheels

...depending on Python version.
theroyallab · Jan 3, 2024 · 2460b2f · 2460b2f
1 parent 451042a
commit 2460b2f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/requirements-amd.txt b/requirements-amd.txt
@@ -3,8 +3,8 @@
 torch
 
 # Exllamav2
-https://github.com/turboderp/exllamav2/releases/download/v0.0.11/exllamav2-0.0.11+rocm5.6-cp311-cp311-linux_x86_64.whl
-https://github.com/turboderp/exllamav2/releases/download/v0.0.11/exllamav2-0.0.11+rocm5.6-cp310-cp310-linux_x86_64.whl
+https://github.com/turboderp/exllamav2/releases/download/v0.0.11/exllamav2-0.0.11+rocm5.6-cp311-cp311-linux_x86_64.whl; python_version == "3.11"
+https://github.com/turboderp/exllamav2/releases/download/v0.0.11/exllamav2-0.0.11+rocm5.6-cp310-cp310-linux_x86_64.whl; python_version == "3.10"
 
 # Pip dependencies
 fastapi
@@ -13,4 +13,4 @@ PyYAML
 progress
 uvicorn
 jinja2 >= 3.0.0
-colorlog
+colorlog