Update app.py

app.py

@@ -29,8 +29,31 @@ if not hf_token:
 # Use the token for authentication
 HfFolder.save_token(hf_token)
 
+# Load the original model to cache
+original_model_id = "stabilityai/stable-video-diffusion-img2vid-xt"
 pipe = StableVideoDiffusionPipeline.from_pretrained(
-    "stabilityai/stable-video-diffusion-img2vid-xt-1-1", torch_dtype=torch.float16, variant="fp16", use_auth_token=hf_token
+    original_model_id,
+    torch_dtype=torch.float16,
+    variant="fp16",
+    use_auth_token=hf_token
+)
+
+# Get the cache directory of the original model
+cache_dir = HfFolder.get_cache_dir()
+model_cache_dir = os.path.join(cache_dir, "diffusers", original_model_id.replace("/", "--"))
+
+# Path to the downloaded svd_xt_1_1.safetensors file
+downloaded_safetensors_path = "svd_xt_1_1.safetensors"
+
+# Replace the original safetensors file with the downloaded one
+original_safetensors_path = os.path.join(model_cache_dir, "svd_xt.safetensors")
+os.replace(downloaded_safetensors_path, original_safetensors_path)
+
+# Load the model from the local directory with the replaced safetensors file
+pipe = StableVideoDiffusionPipeline.from_pretrained(
+    model_cache_dir,
+    torch_dtype=torch.float16,
+    variant="fp16"
 )
 pipe.to("cuda")
 pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)