@@ -502,7 +502,7 @@ def check_inputs(
502502 raise ValueError (f"`max_sequence_length` cannot be greater than 512 but is { max_sequence_length } " )
503503
504504 @staticmethod
505- # Copied from diffusers.pipelines.flux.pipeline_flux.FluxPipeline ._prepare_latent_image_ids
505+ # Copied from diffusers.pipelines.flux.pipeline_flux_utils.FluxMixin ._prepare_latent_image_ids
506506 def _prepare_latent_image_ids (batch_size , height , width , device , dtype ):
507507 latent_image_ids = torch .zeros (height // 2 , width // 2 , 3 )
508508 latent_image_ids [..., 1 ] = latent_image_ids [..., 1 ] + torch .arange (height // 2 )[:, None ]
@@ -517,7 +517,7 @@ def _prepare_latent_image_ids(batch_size, height, width, device, dtype):
517517 return latent_image_ids .to (device = device , dtype = dtype )
518518
519519 @staticmethod
520- # Copied from diffusers.pipelines.flux.pipeline_flux.FluxPipeline ._pack_latents
520+ # Copied from diffusers.pipelines.flux.pipeline_flux_utils.FluxMixin ._pack_latents
521521 def _pack_latents (latents , batch_size , num_channels_latents , height , width ):
522522 latents = latents .view (batch_size , num_channels_latents , height // 2 , 2 , width // 2 , 2 )
523523 latents = latents .permute (0 , 2 , 4 , 1 , 3 , 5 )
0 commit comments