Skip to content

Commit bac8162

Browse files
Wolfgang Romanowskiopenshift-merge-bot[bot]
authored andcommitted
added tensorboard import to CUDA and Rocm training images
1 parent d82a0fa commit bac8162

File tree

4 files changed

+428
-206
lines changed

4 files changed

+428
-206
lines changed

images/runtime/training/cuda/Pipfile

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ torch = "==2.4.1"
1313
sentencepiece = "<0.3,>=0.1.99"
1414
tokenizers = "<1.0,>=0.13.3"
1515
tqdm = "<5.0,>=4.66.2"
16-
trl = ">=0.15.1"
16+
trl = ">=0.15.2"
1717
protobuf = "<6.0.0,>=5.28.0"
1818
simpleeval = "<1.0,>=0.9.13"
1919
safetensors = "*"
@@ -27,6 +27,7 @@ pydantic = ">=2.7.0"
2727
deepspeed = ">=0.14.3"
2828
aiofiles = ">=23.2.1"
2929
async-timeout = "==4.0.3"
30+
tensorboard = "2.19.0"
3031

3132
[dev-packages]
3233

0 commit comments

Comments
 (0)