SherryX Solitude-liu commited on
Commit
304de0b
·
verified ·
1 Parent(s): 5d1d459

Update video_to_video/video_to_video_model.py (#5)

Browse files

- Update video_to_video/video_to_video_model.py (ae5082de621a44c538ccb18fdf258006bef60ff3)


Co-authored-by: Yinhong Liu <[email protected]>

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)