chunzhi/stable-diffusion-cli-on-modal
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Modify Dockerfile and requirements.txt.

Browse Source
This commit is contained in:
hodanov 2023-06-08 21:08:18 +09:00
parent 1b4c76902c
commit 767f9f75fe
2 changed files with 6 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Dockerfile
View File

@ -2,4 +2,4 @@ FROM python:3.11.3-slim-bullseye
COPY requirements.txt / COPY requirements.txt /
RUN apt update \ RUN apt update \
&& apt install -y wget git \ && apt install -y wget git \
&& pip install -r requirements.txt --extra-index-url https://download.pytorch.org/whl/cu117 --pre xformers && pip install -r requirements.txt --extra-index-url https://download.pytorch.org/whl/cu117

12
requirements.txt
View File

@ -1,9 +1,7 @@
accelerate accelerate
scipy diffusers[torch]==0.16.1
diffusers[torch] onnxruntime==1.15.0
safetensors safetensors==0.3.1
torch==2.0.1+cu117 torch==2.0.1+cu117
torchvision transformers==4.29.2
torchmetrics xformers==0.0.20
omegaconf
transformers