smoothieAI commited on
Commit
f187a55
·
verified ·
1 Parent(s): fbc5366

Update pipeline.py

Browse files
Files changed (1) hide show
  1. 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