Yaron Koresh commited on
Commit
f4c39cd
·
verified ·
1 Parent(s): d070b5b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = scheduler2
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()