add bytes and bytes and accelerate for LLMs (#254)

Kaggle · Dec 14, 2023 · 29cd76f · 29cd76f
1 parent e5db47a
commit 29cd76f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docker/gpu.Dockerfile b/docker/gpu.Dockerfile
@@ -27,7 +27,7 @@ RUN npm -v
 
 WORKDIR /usr/src/app/kaggle_environments
 
-RUN pip install Flask
+RUN pip install Flask bitsandbytes accelerate
 
 ADD ./setup.py ./setup.py
 ADD ./README.md ./README.md