Skip to content

Commit 29cd76f

Browse files
authored
add bytes and bytes and accelerate for LLMs (#254)
1 parent e5db47a commit 29cd76f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

docker/gpu.Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ RUN npm -v
2727

2828
WORKDIR /usr/src/app/kaggle_environments
2929

30-
RUN pip install Flask
30+
RUN pip install Flask bitsandbytes accelerate
3131

3232
ADD ./setup.py ./setup.py
3333
ADD ./README.md ./README.md

0 commit comments

Comments
 (0)