@@ -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