Skip to content

Commit 1588e10

Browse files
Add bitsandbytes to ROCm training image
1 parent dea8f17 commit 1588e10

File tree

2 files changed

+11
-1
lines changed

2 files changed

+11
-1
lines changed

images/runtime/training/rocm/Pipfile

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@ verify_ssl = true
99
name = "pytorch"
1010

1111
[packages]
12+
bitsandbytes = ">=0.45.3"
1213
peft = ">=0.14.0"
1314
datasets = ">=2.15.0"
1415
transformers = ">=4.49.0"

images/runtime/training/rocm/Pipfile.lock

Lines changed: 10 additions & 1 deletion
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

0 commit comments

Comments
 (0)