Update Dockerfile

Dockerfile

@@ -13,7 +13,8 @@ ENV TRANSFORMERS_CACHE=/cache \
     HF_HOME=/cache \
     XDG_CACHE_HOME=/cache \
     PYTHONPATH=/code \
-    TORCH_HOME=/cache
+    TORCH_HOME=/cache \
+    BNB_CUDA_VERSION=0
 
 # Install system dependencies
 RUN apt-get update && apt-get install -y \
@@ -37,7 +38,7 @@ RUN pip install --no-cache-dir --upgrade pip && \
     pip install --no-cache-dir 'scikit-learn==1.6.1' && \
     pip install --no-cache-dir 'faiss-cpu==1.7.3' && \
     pip install --no-cache-dir einops>=0.6.1 && \
-    pip install --no-cache-dir bitsandbytes==0.41.3 && \
+    pip install --no-cache-dir -U bitsandbytes>=0.42.0 && \
     pip install --no-cache-dir -r /code/requirements.txt
 
 # Copy the rest of the application