abhishek commited on
Commit
4d7c267
·
1 Parent(s): 614e9d4

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +7 -1
Dockerfile CHANGED
@@ -1,3 +1,9 @@
1
  FROM abhishekkrthakur/diffuzers:latest
 
 
 
2
  RUN chown -R 1000:1000 /app
3
- CMD diffuzers run --model stabilityai/stable-diffusion-2-base --port 7860
 
 
 
 
1
  FROM abhishekkrthakur/diffuzers:latest
2
+ ARG MODEL=https://huggingface.co/stabilityai/stable-diffusion-2-base
3
+ ARG INPAINTING_MODEL=https://huggingface.co/stabilityai/stable-diffusion-2-inpainting
4
+
5
  RUN chown -R 1000:1000 /app
6
+ RUN cd app && git clone ${model} model
7
+ RUN cd app && git clone ${INPAINTING_MODEL} inpainting_model
8
+
9
+ CMD diffuzers run --model model --inpainting_model inpainting_model --port 7860