Skip to content

Commit

Permalink
fix: dependencies
Browse files Browse the repository at this point in the history
  • Loading branch information
mudler committed Aug 8, 2023
1 parent 33afecc commit 5cfa30e
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 5 deletions.
4 changes: 2 additions & 2 deletions Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -37,9 +37,9 @@ ENV PATH /usr/local/cuda/bin:${PATH}
COPY extra/requirements.txt /build/extra/requirements.txt
ENV PATH="/root/.cargo/bin:${PATH}"
RUN pip install --upgrade pip
RUN curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y
RUN if [ "${TARGETARCH}" = "amd64" ]; then \
curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y && \
pip install diffusers invisible_watermark transformers accelerate safetensors;\
pip install git+https://github.com/suno-ai/bark.git diffusers invisible_watermark transformers accelerate safetensors;\
fi
RUN if [ "${BUILD_TYPE}" = "cublas" ] && [ "${TARGETARCH}" = "amd64" ]; then \
pip install torch && pip install auto-gptq;\
Expand Down
4 changes: 1 addition & 3 deletions extra/requirements.txt
Original file line number Diff line number Diff line change
@@ -1,6 +1,4 @@
sentence_transformers
grpcio
google
protobuf
torch
git+https://github.com/suno-ai/bark.git
protobuf

0 comments on commit 5cfa30e

Please sign in to comment.