From ae956e63e0de54502ccf287ecda54c15d9f31ee0 Mon Sep 17 00:00:00 2001
From: Tuuz <24893075+tobycroft@users.noreply.github.com>
Date: Tue, 16 Apr 2024 19:32:25 +0800
Subject: [PATCH] Update app.py

add force_download to make env build more stable
---
 app.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/app.py b/app.py
index 87cf64e..6dda854 100644
--- a/app.py
+++ b/app.py
@@ -49,6 +49,7 @@ def download_model():
             local_dir=CheckpointsDir,
             max_workers=8,
             local_dir_use_symlinks=True,
+            force_download=True, resume_download=False
         )
         # weight
         os.makedirs(f"{CheckpointsDir}/sd-vae-ft-mse/")
@@ -57,6 +58,7 @@ def download_model():
             local_dir=CheckpointsDir+'/sd-vae-ft-mse',
             max_workers=8,
             local_dir_use_symlinks=True,
+            force_download=True, resume_download=False
         )
         #dwpose
         os.makedirs(f"{CheckpointsDir}/dwpose/")
@@ -65,6 +67,7 @@ def download_model():
             local_dir=CheckpointsDir+'/dwpose',
             max_workers=8,
             local_dir_use_symlinks=True,
+            force_download=True, resume_download=False
         )
         #vae
         url = "https://openaipublic.azureedge.net/main/whisper/models/65147644a518d12f04e32d6f3b26facc3f8dd46e5390956a9424a650c0ce22b9/tiny.pt"