File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 8181 """
8282
8383
84- # Copied from diffusers.pipelines.flux.pipeline_flux .calculate_shift
84+ # Copied from diffusers.pipelines.flux.pipeline_flux_utils .calculate_shift
8585def calculate_shift (
8686 image_seq_len ,
8787 base_seq_len : int = 256 ,
Original file line number Diff line number Diff line change 8989"""
9090
9191
92- # Copied from diffusers.pipelines.flux.pipeline_flux .calculate_shift
92+ # Copied from diffusers.pipelines.flux.pipeline_flux_utils .calculate_shift
9393def calculate_shift (
9494 image_seq_len ,
9595 base_seq_len : int = 256 ,
Original file line number Diff line number Diff line change 8686"""
8787
8888
89- # Copied from diffusers.pipelines.flux.pipeline_flux .calculate_shift
89+ # Copied from diffusers.pipelines.flux.pipeline_flux_utils .calculate_shift
9090def calculate_shift (
9191 image_seq_len ,
9292 base_seq_len : int = 256 ,
Original file line number Diff line number Diff line change 6565"""
6666
6767
68- # Copied from diffusers.pipelines.flux.pipeline_flux .calculate_shift
68+ # Copied from diffusers.pipelines.flux.pipeline_flux_utils .calculate_shift
6969def calculate_shift (
7070 image_seq_len ,
7171 base_seq_len : int = 256 ,
Original file line number Diff line number Diff line change 8686"""
8787
8888
89- # Copied from diffusers.pipelines.flux.pipeline_flux .calculate_shift
89+ # Copied from diffusers.pipelines.flux.pipeline_flux_utils .calculate_shift
9090def calculate_shift (
9191 image_seq_len ,
9292 base_seq_len : int = 256 ,
Original file line number Diff line number Diff line change @@ -119,7 +119,7 @@ def forward_with_stg(
119119 return hidden_states
120120
121121
122- # Copied from diffusers.pipelines.flux.pipeline_flux .calculate_shift
122+ # Copied from diffusers.pipelines.flux.pipeline_flux_utils .calculate_shift
123123def calculate_shift (
124124 image_seq_len ,
125125 base_seq_len : int = 256 ,
Original file line number Diff line number Diff line change @@ -124,7 +124,7 @@ def forward_with_stg(
124124 return hidden_states
125125
126126
127- # Copied from diffusers.pipelines.flux.pipeline_flux .calculate_shift
127+ # Copied from diffusers.pipelines.flux.pipeline_flux_utils .calculate_shift
128128def calculate_shift (
129129 image_seq_len ,
130130 base_seq_len : int = 256 ,
Original file line number Diff line number Diff line change 8383"""
8484
8585
86- # Copied from diffusers.pipelines.flux.pipeline_flux .calculate_shift
86+ # Copied from diffusers.pipelines.flux.pipeline_flux_utils .calculate_shift
8787def calculate_shift (
8888 image_seq_len ,
8989 base_seq_len : int = 256 ,
Original file line number Diff line number Diff line change @@ -93,7 +93,7 @@ def compute_empirical_mu(image_seq_len: int, num_steps: int) -> float:
9393 return float (mu )
9494
9595
96- # Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion .retrieve_timesteps
96+ # Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion_utils .retrieve_timesteps
9797def retrieve_timesteps (
9898 scheduler ,
9999 num_inference_steps : Optional [int ] = None ,
@@ -153,7 +153,7 @@ def retrieve_timesteps(
153153 return timesteps , num_inference_steps
154154
155155
156- # Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion_img2img .retrieve_latents
156+ # Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion_utils .retrieve_latents
157157def retrieve_latents (
158158 encoder_output : torch .Tensor , generator : Optional [torch .Generator ] = None , sample_mode : str = "sample"
159159):
Original file line number Diff line number Diff line change 6060"""
6161
6262
63- # Copied from diffusers.pipelines.flux.pipeline_flux .calculate_shift
63+ # Copied from diffusers.pipelines.flux.pipeline_flux_utils .calculate_shift
6464def calculate_shift (
6565 image_seq_len ,
6666 base_seq_len : int = 256 ,
@@ -74,7 +74,7 @@ def calculate_shift(
7474 return mu
7575
7676
77- # Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion .retrieve_timesteps
77+ # Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion_utils .retrieve_timesteps
7878def retrieve_timesteps (
7979 scheduler ,
8080 num_inference_steps : Optional [int ] = None ,
You can’t perform that action at this time.
0 commit comments