Update app.py

app.py

@@ -18,6 +18,11 @@ import random
 from huggingface_hub import login, hf_hub_download
 import spaces
 
+try:
+    hf_hub_download(repo_id="vdo/stable-video-diffusion-img2vid-xt-1-1", local_dir="./checkpoints", cache_dir="./checkpoints")
+except (Exception, BaseException) as error:
+    print(error)
+
 # pipe = StableVideoDiffusionPipeline.from_pretrained(
 #     # "stabilityai/stable-video-diffusion-img2vid-xt-1-1",
 #     "vdo/stable-video-diffusion-img2vid-xt-1-1",
@@ -38,11 +43,20 @@ if not os.path.exists(model_directory):
     )
     pipe.save_pretrained("./checkpoints", variant="fp16")
 else:
-    pipe = StableVideoDiffusionPipeline.from_pretrained(
-        model_directory,
-        torch_dtype=torch.float16,
-        variant="fp16"
-    )
+    try:
+        pipe = StableVideoDiffusionPipeline.from_pretrained(
+            model_directory,
+            torch_dtype=torch.float16,
+            variant="fp16"
+        )
+    except:
+        pipe = StableVideoDiffusionPipeline.from_pretrained(
+            # "stabilityai/stable-video-diffusion-img2vid-xt-1-1",
+            "vdo/stable-video-diffusion-img2vid-xt-1-1",
+            torch_dtype=torch.float16,
+            variant="fp16"
+        )
+        pipe.save_pretrained("./checkpoints", variant="fp16")
 
 pipe.to("cuda")
 # pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)