automatically download model

app.py

@@ -38,18 +38,18 @@ def get_configs(num_steps=1, chopping_size=128, seed=12345):
     print(f'Setting timesteps for inference: {configs.timesteps}')
 
     # path to save noise predictor
-    started_ckpt_path = "noise_predictor_sd_turbo_v5.pth"
-    # started_ckpt_dir = "./weights"
-    # util_common.mkdir(started_ckpt_dir, delete=False, parents=True)
-    # started_ckpt_path = Path(started_ckpt_dir) / started_ckpt_name
-    # if not started_ckpt_path.exists():
-        # load_file_from_url(
-            # url="https://huggingface.co/OAOA/InvSR/resolve/main/noise_predictor_sd_turbo_v5.pth",
-            # model_dir=started_ckpt_dir,
-            # progress=True,
-            # file_name=started_ckpt_name,
-        # )
-    configs.model_start.ckpt_path = started_ckpt_path
+    started_ckpt_name = "noise_predictor_sd_turbo_v5.pth"
+    started_ckpt_dir = "./weights"
+    util_common.mkdir(started_ckpt_dir, delete=False, parents=True)
+    started_ckpt_path = Path(started_ckpt_dir) / started_ckpt_name
+    if not started_ckpt_path.exists():
+        load_file_from_url(
+            url="https://huggingface.co/OAOA/InvSR/resolve/main/noise_predictor_sd_turbo_v5.pth",
+            model_dir=started_ckpt_dir,
+            progress=True,
+            file_name=started_ckpt_name,
+        )
+    configs.model_start.ckpt_path = str(started_ckpt_path)
 
     configs.bs = 1
     configs.seed = 12345