Merge remote-tracking branch 'upstream/main'

red-hat-data-services · Mar 1, 2025 · 97f1712 · 97f1712
2 parents 62c8665 + 1588e10
commit 97f1712
Show file tree

Hide file tree

Showing 5 changed files with 23 additions and 3 deletions.
diff --git a/images/runtime/training/cuda/Pipfile b/images/runtime/training/cuda/Pipfile
@@ -4,6 +4,7 @@ verify_ssl = true
 name = "pypi"
 
 [packages]
+bitsandbytes = ">=0.45.3"
 peft = ">=0.14.0"
 datasets = ">=2.15.0"
 transformers = ">=4.49.0"

diff --git a/images/runtime/training/cuda/Pipfile.lock b/images/runtime/training/cuda/Pipfile.lock
diff --git a/images/runtime/training/rocm/Dockerfile b/images/runtime/training/rocm/Dockerfile
@@ -73,7 +73,7 @@ RUN export TMP_DIR=$(mktemp -d) \
     && cd $TMP_DIR \
     && git clone --depth 1 --branch v2.7.4 https://github.com/Dao-AILab/flash-attention.git \
     && cd flash-attention \
-    && git submodule update --init
+    && git submodule update --init \
     && MAX_JOBS="16" python3 setup.py install --verbose \
     && rm -rf $TMP_DIR
 

diff --git a/images/runtime/training/rocm/Pipfile b/images/runtime/training/rocm/Pipfile
@@ -9,6 +9,7 @@ verify_ssl = true
 name = "pytorch"
 
 [packages]
+bitsandbytes = ">=0.45.3"
 peft = ">=0.14.0"
 datasets = ">=2.15.0"
 transformers = ">=4.49.0"

diff --git a/images/runtime/training/rocm/Pipfile.lock b/images/runtime/training/rocm/Pipfile.lock