test:bnb

Signed-off-by: Will Johnson <mwjohnson728@gmail.com>
foundation-model-stack · Jan 2, 2025 · 443b6b5 · 443b6b5
1 parent c746655
commit 443b6b5
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/build/Dockerfile b/build/Dockerfile
@@ -150,6 +150,8 @@ RUN if [[ "${ENABLE_FMS_ACCELERATION}" == "true" ]]; then \
         python -m fms_acceleration.cli install fms_acceleration_moe; \
     fi
 
+RUN pip install bitsandbytes==0.43.3
+
 RUN if [[ "${ENABLE_AIM}" == "true" ]]; then \
         python -m pip install --user "$(head bdist_name)[aim]"; \
     fi