Update pipeline.py
Browse files- pipeline.py +2 -2
pipeline.py
CHANGED
@@ -1333,8 +1333,8 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
|
|
1333 |
)
|
1334 |
elif(latent_mode == "video"):
|
1335 |
# 4. Prepare timesteps
|
1336 |
-
timesteps, num_inference_steps = retrieve_timesteps(self.scheduler, num_inference_steps, device, timesteps)
|
1337 |
-
timesteps, num_inference_steps = self.get_timesteps(num_inference_steps, init_image_strength, device)
|
1338 |
latent_timestep = timesteps[:1].repeat(batch_size * num_videos_per_prompt)
|
1339 |
self._num_timesteps = len(timesteps)
|
1340 |
num_channels_latents = self.unet.config.in_channels
|
|
|
1333 |
)
|
1334 |
elif(latent_mode == "video"):
|
1335 |
# 4. Prepare timesteps
|
1336 |
+
# timesteps, num_inference_steps = retrieve_timesteps(self.scheduler, num_inference_steps, device, timesteps)
|
1337 |
+
# timesteps, num_inference_steps = self.get_timesteps(num_inference_steps, init_image_strength, device)
|
1338 |
latent_timestep = timesteps[:1].repeat(batch_size * num_videos_per_prompt)
|
1339 |
self._num_timesteps = len(timesteps)
|
1340 |
num_channels_latents = self.unet.config.in_channels
|