Skip to content

Commit

Permalink
Modify modules versions for linux/arm64
Browse files Browse the repository at this point in the history
  • Loading branch information
whitehara committed Feb 25, 2024
1 parent e838927 commit b48ff50
Show file tree
Hide file tree
Showing 3 changed files with 7 additions and 8 deletions.
1 change: 1 addition & 0 deletions Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,7 @@ RUN apt-get update -y && \
COPY requirements_docker.txt requirements_versions.txt /tmp/
RUN pip install --no-cache-dir -r /tmp/requirements_docker.txt -r /tmp/requirements_versions.txt && \
rm -f /tmp/requirements_docker.txt /tmp/requirements_versions.txt
RUN pip install --no-cache-dir xformers==0.0.22 --no-dependencies
RUN curl -fsL -o /usr/local/lib/python3.10/dist-packages/gradio/frpc_linux_amd64_v0.2 https://cdn-media.huggingface.co/frpc-gradio-0.2/frpc_linux_amd64 && \
chmod +x /usr/local/lib/python3.10/dist-packages/gradio/frpc_linux_amd64_v0.2

Expand Down
2 changes: 1 addition & 1 deletion docker.md
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
# Fooocus on Docker

The docker image is based on NVIDIA CUDA 12.3 and PyTorch 2.1, see [Dockerfile](Dockerfile) and [requirements_docker.txt](requirements_docker.txt) for details.
The docker image is based on NVIDIA CUDA 12.3 and PyTorch 2.0, see [Dockerfile](Dockerfile) and [requirements_docker.txt](requirements_docker.txt) for details.

## Quick start

Expand Down
12 changes: 5 additions & 7 deletions requirements_docker.txt
Original file line number Diff line number Diff line change
@@ -1,7 +1,5 @@
torch==2.2.0
torchvision==0.17.0
torchaudio==2.2.0
torchtext==0.17.0
torchdata==0.7.1
xformers==0.0.24
triton==2.2.0
torch==2.0.1
torchvision==0.15.2
torchaudio==2.0.2
torchtext==0.15.2
torchdata==0.6.1

0 comments on commit b48ff50

Please sign in to comment.