testbot commited on
Commit
95c4e87
1 Parent(s): fec80ac

add accelerate as a deps

Browse files
Files changed (1) hide show
  1. 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