From 443b6b5159805805038f524b29362af421313226 Mon Sep 17 00:00:00 2001
From: Will Johnson <mwjohnson728@gmail.com>
Date: Thu, 2 Jan 2025 12:05:05 -0500
Subject: [PATCH] test:bnb

Signed-off-by: Will Johnson <mwjohnson728@gmail.com>
---
 build/Dockerfile | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/build/Dockerfile b/build/Dockerfile
index e12f1de48..f61b91418 100644
--- 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