envs

app.py

@@ -89,28 +89,37 @@ If you have any questions, please feel free to reach me out at <b>ywl@stu.pku.ed
 
 os.makedirs("models/personalized")
 os.makedirs("models/sd1-5")
-os.system(f'wget https://huggingface.co/TencentARC/ImageConductor/resolve/main/flow_controlnet.ckpt?download=true -P models/')
-os.system(f'mv models/flow_controlnet.ckpt?download=true models/flow_controlnet.ckpt')
 
-os.system(f'wget https://huggingface.co/TencentARC/ImageConductor/resolve/main/image_controlnet.ckpt?download=true -P models/')
-os.system(f'mv models/image_controlnet.ckpt?download=true models/image_controlnet.ckpt')
+if not os.path.exists("models/flow_controlnet.ckpt"):
+    os.system(f'wget https://huggingface.co/TencentARC/ImageConductor/resolve/main/flow_controlnet.ckpt?download=true -P models/')
+    os.system(f'mv models/flow_controlnet.ckpt?download=true models/flow_controlnet.ckpt')
 
-os.system(f'wget https://huggingface.co/TencentARC/ImageConductor/resolve/main/unet.ckpt?download=true -P models/')
-os.system(f'mv models/unet.ckpt?download=true models/unet.ckpt')
+if not os.path.exists("models/image_controlnet.ckpt"):
+    os.system(f'wget https://huggingface.co/TencentARC/ImageConductor/resolve/main/image_controlnet.ckpt?download=true -P models/')
+    os.system(f'mv models/image_controlnet.ckpt?download=true models/image_controlnet.ckpt')
 
 
-os.system(f'wget https://huggingface.co/runwayml/stable-diffusion-v1-5/resolve/main/unet/config.json?download=true -P models/sd1-5/')
-os.system(f'mv models/sd1-5/config.json?download=true  models/sd1-5/config.json')
+if not os.path.exists("models/unet.ckpt"):
+    os.system(f'wget https://huggingface.co/TencentARC/ImageConductor/resolve/main/unet.ckpt?download=true -P models/')
+    os.system(f'mv models/unet.ckpt?download=true models/unet.ckpt')
 
-os.system(f'cp -r models/unet.ckpt  models/sd1-5/unet.ckpt')
+if not os.path.exists("models/sd1-5/config.json"):
+    os.system(f'wget https://huggingface.co/runwayml/stable-diffusion-v1-5/resolve/main/unet/config.json?download=true -P models/sd1-5/')
+    os.system(f'mv models/sd1-5/config.json?download=true  models/sd1-5/config.json')
+
+if not os.path.exists("models/sd1-5/unet.ckpt"):
+    os.system(f'cp -r models/unet.ckpt  models/sd1-5/unet.ckpt')
 
 # os.system(f'wget https://huggingface.co/runwayml/stable-diffusion-v1-5/resolve/main/unet/diffusion_pytorch_model.bin?download=true -P models/sd1-5/')
 
-os.system(f'wget https://huggingface.co/TencentARC/ImageConductor/resolve/main/helloobjects_V12c.safetensors?download=true -P models/personalized')
-os.system(f'mv models/personalized/helloobjects_V12c.safetensors?download=true models/personalized/helloobjects_V12c.safetensors')
+if not os.path.exists("models/personalized/helloobjects_V12c.safetensors"):
+    os.system(f'wget https://huggingface.co/TencentARC/ImageConductor/resolve/main/helloobjects_V12c.safetensors?download=true -P models/personalized')
+    os.system(f'mv models/personalized/helloobjects_V12c.safetensors?download=true models/personalized/helloobjects_V12c.safetensors')
+
 
-os.system(f'wget https://huggingface.co/TencentARC/ImageConductor/resolve/main/TUSUN.safetensors?download=true -P models/personalized')
-os.system(f'mv models/personalized/TUSUN.safetensors?download=true models/personalized/TUSUN.safetensors')
+if not os.path.exists("models/personalized/TUSUN.safetensors"):
+    os.system(f'wget https://huggingface.co/TencentARC/ImageConductor/resolve/main/TUSUN.safetensors?download=true -P models/personalized')
+    os.system(f'mv models/personalized/TUSUN.safetensors?download=true models/personalized/TUSUN.safetensors')