Spaces:
Sleeping
Sleeping
Yaron Koresh
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -129,7 +129,7 @@ video_pipe = CogVideoXImageToVideoPipeline.from_pretrained(
|
|
129 |
"THUDM/CogVideoX-5b-I2V",
|
130 |
torch_dtype=torch.bfloat16
|
131 |
).to(device)
|
132 |
-
video_pipe.scheduler =
|
133 |
|
134 |
video_pipe.vae.enable_tiling()
|
135 |
video_pipe.vae.enable_slicing()
|
|
|
129 |
"THUDM/CogVideoX-5b-I2V",
|
130 |
torch_dtype=torch.bfloat16
|
131 |
).to(device)
|
132 |
+
video_pipe.scheduler = scheduler
|
133 |
|
134 |
video_pipe.vae.enable_tiling()
|
135 |
video_pipe.vae.enable_slicing()
|