Update dockerfile

parzival418 · parzival418 · commit 67ca07e66aa6 · 2022-12-19T09:40:31.000-08:00
diff --git a/Dockerfile b/Dockerfile
@@ -103,7 +103,7 @@ ENV HF_AUTH_TOKEN=${HF_AUTH_TOKEN}
 # "CompVis/stable-diffusion-v1-4", "hakurei/waifu-diffusion",
 # "stabilityai/stable-diffusion-2",
 # "stabilityai/stable-diffusion-2-inpainting" etc.
-ARG MODEL_ID="spaceinvader/tarrot"
+ARG MODEL_ID
 ENV MODEL_ID=${MODEL_ID}
 
 # "" = model default.
@@ -132,9 +132,9 @@ ARG MODEL_URL=""
 ENV MODEL_URL=${MODEL_URL}
 # If set, it will be downloaded and converted to diffusers format, and
 # saved in a directory with same MODEL_ID name to be loaded by diffusers.
-ARG CHECKPOINT_URL="https://huggingface.co/spaceinvader/tarrot/resolve/main/tarot_rws_step_2500.ckpt"
+ARG CHECKPOINT_URL
 ENV CHECKPOINT_URL=${CHECKPOINT_URL}
-ARG CHECKPOINT_CONFIG_URL="https://raw.githubusercontent.com/Stability-AI/stablediffusion/main/configs/stable-diffusion/v2-inference-v.yaml"
+ARG CHECKPOINT_CONFIG_URL
 ENV CHECKPOINT_CONFIG_URL=${CHECKPOINT_CONFIG_URL}
 # Set to true to NOT download model at build time, rather at init / usage.
 ARG RUNTIME_DOWNLOADS=0
