fix aqlm version

Former-commit-id: 05673f81f0295c76957f3247c62f95fda322a63e
2024-03-09 00:09:09 +08:00
parent 53ab28533e
commit 9b97b23ce7
6 changed files with 8 additions and 18 deletions
--- a/setup.py
+++ b/setup.py
@@ -26,7 +26,7 @@ extra_require = {
    "bitsandbytes": ["bitsandbytes>=0.39.0"],
    "gptq": ["optimum>=1.16.0", "auto-gptq>=0.5.0"],
    "awq": ["autoawq"],
-    "aqlm": ["aqlm[gpu,cpu]"],
+    "aqlm": ["aqlm[gpu]>=1.1.0", "transformers @ git+https://github.com/huggingface/transformers.git"],
    "galore": ["galore_torch @ git+https://github.com/hiyouga/GaLore.git"],
    "qwen": ["tiktoken", "transformers_stream_generator"],
 }