Skip to content

Commit a0ffed1

Browse files
authored
Update src/diffusers/pipelines/flux/pipeline_flux_fill.py
1 parent f0fac62 commit a0ffed1

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

src/diffusers/pipelines/flux/pipeline_flux_fill.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -902,10 +902,10 @@ def __call__(
902902
image_seq_len = (int(height) // self.vae_scale_factor // 2) * (int(width) // self.vae_scale_factor // 2)
903903
mu = calculate_shift(
904904
image_seq_len,
905-
self.scheduler.config.base_image_seq_len,
906-
self.scheduler.config.max_image_seq_len,
907-
self.scheduler.config.base_shift,
908-
self.scheduler.config.max_shift,
905+
self.scheduler.config.get("base_image_seq_len", 256),
906+
self.scheduler.config.get("max_image_seq_len", 4096),
907+
self.scheduler.config.get("base_shift", 0.5),
908+
self.scheduler.config.get("max_shift", 1.15),
909909
)
910910
timesteps, num_inference_steps = retrieve_timesteps(
911911
self.scheduler,

0 commit comments

Comments
 (0)