Tonic commited on
Commit
57eee7d
β€’
1 Parent(s): 0ad3cbd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -26,11 +26,11 @@ pipe = StableVideoDiffusionPipeline.from_pretrained(
26
  variant="fp16",
27
  )
28
 
29
- downloaded_safetensors_path = "./svd_xt_1_1.safetensors"
30
  if not os.path.exists(downloaded_safetensors_path):
31
  raise FileNotFoundError(f"The file {downloaded_safetensors_path} was not found.")
32
 
33
- cache_dir = os.path.join(os.path.expanduser("~"), "./cache/")
34
  model_cache_dir = os.path.join(cache_dir, original_model_id.replace("/", "--"))
35
  original_safetensors_path = os.path.join(model_cache_dir, "svd_xt.safetensors")
36
  os.replace(downloaded_safetensors_path, original_safetensors_path)
 
26
  variant="fp16",
27
  )
28
 
29
+ downloaded_safetensors_path = "svd_xt_1_1.safetensors"
30
  if not os.path.exists(downloaded_safetensors_path):
31
  raise FileNotFoundError(f"The file {downloaded_safetensors_path} was not found.")
32
 
33
+ cache_dir = os.path.join(os.path.expanduser("~"), "cache")
34
  model_cache_dir = os.path.join(cache_dir, original_model_id.replace("/", "--"))
35
  original_safetensors_path = os.path.join(model_cache_dir, "svd_xt.safetensors")
36
  os.replace(downloaded_safetensors_path, original_safetensors_path)