up

sayakpaul · sayakpaul · commit 9f9eb6ae5b73 · 2025-12-11T10:46:23.000+05:30
diff --git a/src/diffusers/modular_pipelines/flux2/encoders.py b/src/diffusers/modular_pipelines/flux2/encoders.py
@@ -43,7 +43,7 @@ def format_text_input(prompts: List[str], system_message: str = None):
     ]
 
 
-# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion_img2img.retrieve_latents
+# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion_utils.retrieve_latents
 def retrieve_latents(
     encoder_output: torch.Tensor, generator: Optional[torch.Generator] = None, sample_mode: str = "sample"
 ):
diff --git a/src/diffusers/modular_pipelines/z_image/encoders.py b/src/diffusers/modular_pipelines/z_image/encoders.py
@@ -80,7 +80,7 @@ def get_qwen_prompt_embeds(
     return prompt_embeds_list
 
 
-# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion_img2img.retrieve_latents
+# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion_utils.retrieve_latents
 def retrieve_latents(
     encoder_output: torch.Tensor, generator: Optional[torch.Generator] = None, sample_mode: str = "sample"
 ):

Original file line number	Diff line number	Diff line change
`@@ -43,7 +43,7 @@ def format_text_input(prompts: List[str], system_message: str = None):`
`43`	`43`	`]`
`44`	`44`
`45`	`45`
`46`		`-# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion_img2img.retrieve_latents`
	`46`	`+# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion_utils.retrieve_latents`
`47`	`47`	`def retrieve_latents(`
`48`	`48`	`encoder_output: torch.Tensor, generator: Optional[torch.Generator] = None, sample_mode: str = "sample"`
`49`	`49`	`):`