add base docker image with pytorch 2.7.0 and variant for cuda 12.8 (#2551)

* add base docker image with pytorch 2.7.0 and variant for cuda 12.8

* my bash is terrible
This commit is contained in:
Wing Lian
2025-04-23 14:59:03 -04:00
committed by GitHub
parent c4053481ff
commit 0d691cc2a7
3 changed files with 17 additions and 0 deletions

View File

@@ -19,6 +19,7 @@ datasets==3.5.0
deepspeed>=0.15.4
trl==0.16.1
hf_xet==1.0.0
hqq==0.2.5
optimum==1.16.2
hf_transfer