Skip to content

Commit 93f7a7f

Browse files
committed
fix
1 parent f52a08b commit 93f7a7f

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/diffusers/pipelines/ltx/pipeline_ltx_condition.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1067,7 +1067,6 @@ def __call__(
10671067
timesteps = sigmas * 1000
10681068
timesteps, num_inference_steps = retrieve_timesteps(self.scheduler, num_inference_steps, device, timesteps)
10691069
num_warmup_steps = max(len(timesteps) - num_inference_steps * self.scheduler.order, 0)
1070-
self._num_timesteps = len(timesteps)
10711070

10721071
latent_sigma = None
10731072
if denoise_strength < 1:
@@ -1076,6 +1075,8 @@ def __call__(
10761075
)
10771076
latent_sigma = sigmas[:1].repeat(batch_size * num_videos_per_prompt)
10781077

1078+
self._num_timesteps = len(timesteps)
1079+
10791080
# 5. Prepare latent variables
10801081
num_channels_latents = self.transformer.config.in_channels
10811082
latents, conditioning_mask, video_coords, extra_conditioning_num_latents = self.prepare_latents(

0 commit comments

Comments
 (0)