Merge remote-tracking branch 'upstream/main'

Author: dchourasia

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"

images/runtime/training/cuda/Pipfile.lock

@@ -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
 

images/runtime/training/rocm/Dockerfile

@@ -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"