Skip to content

Commit 806d6bc

Browse files
Upgrade HuggingFace libraries in KFTO ROCm PyTorch 2.5 image
1 parent 643ad20 commit 806d6bc

File tree

2 files changed

+770
-706
lines changed

2 files changed

+770
-706
lines changed

images/runtime/training/py311-rocm62-torch251/Pipfile

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -9,18 +9,18 @@ verify_ssl = true
99
name = "pytorch"
1010

1111
[packages]
12-
peft = ">=0.14.0"
12+
peft = ">=0.15.2"
1313
datasets = ">=2.15.0"
1414
liger-kernel = "==0.5.4"
15-
transformers = ">=4.49.0"
15+
transformers = ">=4.51.3"
1616
numpy = "<2.0.0,>=1.23.5"
17-
accelerate = ">=1.4.0"
17+
accelerate = ">=1.6.0"
1818
torch = {version = "==2.5.1+rocm6.2", index = "pytorch"}
1919
pytorch-triton-rocm = {version = "*", index = "pytorch"}
2020
sentencepiece = "<0.3,>=0.1.99"
2121
tokenizers = "<1.0,>=0.13.3"
2222
tqdm = "<5.0,>=4.66.2"
23-
trl = ">=0.15.2"
23+
trl = ">=0.17.0"
2424
protobuf = "<6.0.0,>=5.28.0"
2525
simpleeval = "<1.0,>=0.9.13"
2626
safetensors = "*"

0 commit comments

Comments
 (0)