Update app.py

app.py

@@ -23,9 +23,10 @@ import uuid
 import random
 from huggingface_hub import hf_hub_download
 
-hf_hub_download(repo_id="stabilityai/stable-video-diffusion-img2vid-xt", filename="svd_xt.safetensors", local_dir="checkpoints") 
+hf_hub_download(repo_id="stabilityai/stable-video-diffusion-img2vid-xt", filename="svd_xt.safetensors", local_dir="/workspace/checkpoints") 
+hf_hub_download(repo_id="vdo/stable-video-diffusion-img2vid-xt-1-1", filename="svd_xt_1_1.safetensors", local_dir="/workspace/checkpoints") 
 
-version = "svd_xt"
+version = "svd_xt_1_1"
 device = "cuda"
 max_64_bit_int = 2**63 - 1
 
@@ -58,6 +59,10 @@ if version == "svd_xt":
     num_frames = 25
     num_steps = 30
     model_config = "scripts/sampling/configs/svd_xt.yaml"
+elif version == "svd_xt_1_1":
+    num_frames = 25
+    num_steps = 30
+    model_config = "scripts/sampling/configs/svd_xt_1_1.yaml"
 else:
     raise ValueError(f"Version {version} does not exist.")