add 12.8.1 cuda to the base matrix (#2426)

* add 12.8.1 cuda to the base matrix

* use nightly

* bump deepspeed and set no binary

* deepspeed binary fixes hopefully

* install deepspeed by itself

* multiline fix

* make sure ninja is installed

* try with reversion of packaging/setuptools/wheel install

* use license instead of license-file

* try rolling back packaging and setuptools versions

* comment out license for validation for now

* make sure packaging version is consistent

* more parity across tests and docker images for packaging/setuptools
This commit is contained in:
Wing Lian
2025-03-21 10:17:25 -04:00
committed by GitHub
parent 38df5a36ea
commit aae4337f40
11 changed files with 60 additions and 15 deletions

View File

@@ -1,7 +1,7 @@
--extra-index-url https://huggingface.github.io/autogptq-index/whl/cu118/
# START section of dependencies that don't install on Darwin/MacOS
bitsandbytes==0.45.2
bitsandbytes==0.45.3
triton>=3.0.0
mamba-ssm==1.2.0.post1
flash-attn==2.7.4.post1
@@ -10,14 +10,14 @@ autoawq==0.2.7.post3
liger-kernel==0.5.3
# END section
packaging==24.2
packaging==23.2
peft==0.15.0
transformers==4.49.0
tokenizers>=0.21.1
accelerate==1.5.2
datasets==3.4.1
deepspeed==0.16.1
deepspeed==0.16.4
trl==0.15.1
optimum==1.16.2