Skip to content

Commit dea8f17

Browse files
astefanuttiopenshift-merge-bot[bot]
astefanutti
authored and
openshift-merge-bot[bot]
committed
Add bitsandbytes to CUDA training image
1 parent 8ec4e0c commit dea8f17

File tree

2 files changed

+11
-1
lines changed

2 files changed

+11
-1
lines changed

‎images/runtime/training/cuda/Pipfile

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@ verify_ssl = true
44
name = "pypi"
55

66
[packages]
7+
bitsandbytes = ">=0.45.3"
78
peft = ">=0.14.0"
89
datasets = ">=2.15.0"
910
transformers = ">=4.49.0"

‎images/runtime/training/cuda/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)