We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 643ad20 commit 806d6bcCopy full SHA for 806d6bc
images/runtime/training/py311-rocm62-torch251/Pipfile
@@ -9,18 +9,18 @@ verify_ssl = true
9
name = "pytorch"
10
11
[packages]
12
-peft = ">=0.14.0"
+peft = ">=0.15.2"
13
datasets = ">=2.15.0"
14
liger-kernel = "==0.5.4"
15
-transformers = ">=4.49.0"
+transformers = ">=4.51.3"
16
numpy = "<2.0.0,>=1.23.5"
17
-accelerate = ">=1.4.0"
+accelerate = ">=1.6.0"
18
torch = {version = "==2.5.1+rocm6.2", index = "pytorch"}
19
pytorch-triton-rocm = {version = "*", index = "pytorch"}
20
sentencepiece = "<0.3,>=0.1.99"
21
tokenizers = "<1.0,>=0.13.3"
22
tqdm = "<5.0,>=4.66.2"
23
-trl = ">=0.15.2"
+trl = ">=0.17.0"
24
protobuf = "<6.0.0,>=5.28.0"
25
simpleeval = "<1.0,>=0.9.13"
26
safetensors = "*"
0 commit comments