Spaces:
Configuration error
Configuration error
Update video_to_video/video_to_video_model.py
Browse files
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 |
-
"/
|
| 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)
|