Skip to content

Commit 00e9670

Browse files
yiyixuxua-r-r-o-w
andauthored
Apply suggestions from code review
Co-authored-by: Aryan <[email protected]>
1 parent 64df9af commit 00e9670

File tree

1 file changed

+0
-6
lines changed

1 file changed

+0
-6
lines changed

src/diffusers/pipelines/ltx/pipeline_ltx_condition.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -584,8 +584,6 @@ def add_noise_to_image_conditioning_latents(
584584
Add timestep-dependent noise to the hard-conditioning latents. This helps with motion continuity, especially
585585
when conditioned on a single frame.
586586
"""
587-
# YiYi TODO: testing only, remove this change before merging
588-
# generator = torch.Generator(device="cpu").manual_seed(0)
589587
noise = randn_tensor(
590588
latents.shape,
591589
generator=generator,
@@ -619,8 +617,6 @@ def prepare_latents(
619617

620618
shape = (batch_size, num_channels_latents, num_latent_frames, latent_height, latent_width)
621619
latents = randn_tensor(shape, generator=generator, device=device, dtype=dtype)
622-
# YiYi TODO: testing only, remove this change before merging
623-
# latents = torch.load("/raid/yiyi/LTX-Video/init_latents.pt").to(device, dtype=dtype)
624620

625621
condition_latent_frames_mask = torch.zeros((batch_size, num_latent_frames), device=device, dtype=torch.float32)
626622

@@ -662,8 +658,6 @@ def prepare_latents(
662658
condition_latents = condition_latents[:, :, :num_prefix_latent_frames]
663659

664660
noise = randn_tensor(condition_latents.shape, generator=generator, device=device, dtype=dtype)
665-
# YiYi TODO: testing only, remove this change before merging
666-
# noise = torch.load("/raid/yiyi/LTX-Video/noise.pt").to(device, dtype=dtype)
667661
condition_latents = torch.lerp(noise, condition_latents, strength)
668662

669663
condition_video_ids = self._prepare_video_ids(

0 commit comments

Comments
 (0)