Update setup.py

Former-commit-id: 4c017fe014b708d79c65eff24329b9c324399461
This commit is contained in:
hoshi-hiyouga 2024-10-07 16:26:50 +08:00 committed by GitHub
parent 2b69ae0eb2
commit 6e111eb29f

View File

@ -54,7 +54,7 @@ extra_require = {
"gptq": ["optimum>=1.17.0", "auto-gptq>=0.5.0"],
"awq": ["autoawq"],
"aqlm": ["aqlm[gpu]>=1.1.0"],
"vllm": ["vllm>=0.4.3,<=0.6.3"],
"vllm": ["vllm>=0.4.3,<=0.6.2"],
"galore": ["galore-torch"],
"badam": ["badam>=1.2.1"],
"adam-mini": ["adam-mini"],