Update video_to_video/video_to_video_model.py

#5
video_to_video/video_to_video_model.py CHANGED
@@ -80,7 +80,7 @@ class VideoToVideo_sr():
80
 
81
  # Temporal VAE
82
  vae = AutoencoderKLTemporalDecoder.from_pretrained(
83
- "/home/test/Workspace/yhliu/VSR/ours/checkpoints/stable-video-diffusion-img2vid", subfolder="vae", variant="fp16"
84
  )
85
  vae.eval()
86
  vae.requires_grad_(False)
 
80
 
81
  # Temporal VAE
82
  vae = AutoencoderKLTemporalDecoder.from_pretrained(
83
+ "stabilityai/stable-video-diffusion-img2vid", subfolder="vae", variant="fp16"
84
  )
85
  vae.eval()
86
  vae.requires_grad_(False)