Spaces:
Runtime error
Runtime error
testbot
commited on
Commit
•
95c4e87
1
Parent(s):
fec80ac
add accelerate as a deps
Browse files- Dockerfile +2 -1
Dockerfile
CHANGED
@@ -21,7 +21,8 @@ RUN pip install --no-cache-dir pip && \
|
|
21 |
gradio \
|
22 |
torch \
|
23 |
numpy \
|
24 |
-
transformers
|
|
|
25 |
|
26 |
# Prepare bloomz.cpp
|
27 |
RUN git clone https://github.com/NouamaneTazi/bloomz.cpp
|
|
|
21 |
gradio \
|
22 |
torch \
|
23 |
numpy \
|
24 |
+
transformers \
|
25 |
+
accelerate
|
26 |
|
27 |
# Prepare bloomz.cpp
|
28 |
RUN git clone https://github.com/NouamaneTazi/bloomz.cpp
|