Skip to content

Commit 3d875cb

Browse files
committed
make fix-copies.
1 parent a92ed87 commit 3d875cb

4 files changed

Lines changed: 9 additions & 9 deletions

File tree

src/diffusers/pipelines/cosmos/pipeline_cosmos2_5_predict.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@ def __init__(self, *args, **kwargs):
5353
logger = logging.get_logger(__name__) # pylint: disable=invalid-name
5454

5555

56-
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion_img2img.retrieve_latents
56+
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion_utils.retrieve_latents
5757
def retrieve_latents(
5858
encoder_output: torch.Tensor, generator: Optional[torch.Generator] = None, sample_mode: str = "sample"
5959
):

src/diffusers/pipelines/z_image/pipeline_z_image_controlnet.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -98,7 +98,7 @@
9898
"""
9999

100100

101-
# Copied from diffusers.pipelines.flux.pipeline_flux.calculate_shift
101+
# Copied from diffusers.pipelines.flux.pipeline_flux_utils.calculate_shift
102102
def calculate_shift(
103103
image_seq_len,
104104
base_seq_len: int = 256,
@@ -112,7 +112,7 @@ def calculate_shift(
112112
return mu
113113

114114

115-
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion_img2img.retrieve_latents
115+
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion_utils.retrieve_latents
116116
def retrieve_latents(
117117
encoder_output: torch.Tensor, generator: Optional[torch.Generator] = None, sample_mode: str = "sample"
118118
):
@@ -126,7 +126,7 @@ def retrieve_latents(
126126
raise AttributeError("Could not access latents of provided encoder_output")
127127

128128

129-
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion.retrieve_timesteps
129+
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion_utils.retrieve_timesteps
130130
def retrieve_timesteps(
131131
scheduler,
132132
num_inference_steps: Optional[int] = None,

src/diffusers/pipelines/z_image/pipeline_z_image_controlnet_inpaint.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -98,7 +98,7 @@
9898
"""
9999

100100

101-
# Copied from diffusers.pipelines.flux.pipeline_flux.calculate_shift
101+
# Copied from diffusers.pipelines.flux.pipeline_flux_utils.calculate_shift
102102
def calculate_shift(
103103
image_seq_len,
104104
base_seq_len: int = 256,
@@ -112,7 +112,7 @@ def calculate_shift(
112112
return mu
113113

114114

115-
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion_img2img.retrieve_latents
115+
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion_utils.retrieve_latents
116116
def retrieve_latents(
117117
encoder_output: torch.Tensor, generator: Optional[torch.Generator] = None, sample_mode: str = "sample"
118118
):
@@ -126,7 +126,7 @@ def retrieve_latents(
126126
raise AttributeError("Could not access latents of provided encoder_output")
127127

128128

129-
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion.retrieve_timesteps
129+
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion_utils.retrieve_timesteps
130130
def retrieve_timesteps(
131131
scheduler,
132132
num_inference_steps: Optional[int] = None,

src/diffusers/pipelines/z_image/pipeline_z_image_omni.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,7 @@
6161
"""
6262

6363

64-
# Copied from diffusers.pipelines.flux.pipeline_flux.calculate_shift
64+
# Copied from diffusers.pipelines.flux.pipeline_flux_utils.calculate_shift
6565
def calculate_shift(
6666
image_seq_len,
6767
base_seq_len: int = 256,
@@ -75,7 +75,7 @@ def calculate_shift(
7575
return mu
7676

7777

78-
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion.retrieve_timesteps
78+
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion_utils.retrieve_timesteps
7979
def retrieve_timesteps(
8080
scheduler,
8181
num_inference_steps: Optional[int] = None,

0 commit comments

Comments
 (0)