Skip to content

Commit 836a7b3

Browse files
JasonHokuclaude
andcommitted
fix: distributed worker missing 8 sampling parameters causing different outputs
The worker's generate_image() call was only passing basic parameters (seed, steps, cfg, sampler, scheduler, denoise, attention_mode) but missing all advanced/model-specific parameters that the local path passes. This caused workers to produce different images than local runs. Added missing parameters: - model_sampling_override (noise schedule for Flux/SD3/AuraFlow) - model_sampling_shift (shift values for model sampling) - model_sampling_flux_max_shift / model_sampling_flux_base_shift - use_advanced_sampling (SamplerCustomAdvanced vs standard KSampler) - advanced_guider / advanced_scheduler - flux_guidance_value - width / height (for Flux2Scheduler computation) All parameters were already present in the job config dict from expand_configs() but were never forwarded to generate_image(). Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
1 parent afd1c07 commit 836a7b3

1 file changed

Lines changed: 11 additions & 1 deletion

File tree

distribution_worker.py

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -593,7 +593,17 @@ def _process_job(self, job):
593593
self._patched_model, config["seed"], config["steps"], config["cfg"],
594594
config["sampler"], config["scheduler"], pos_cond, neg_cond,
595595
latent_in, config["denoise"],
596-
attention_mode=attention_mode
596+
attention_mode=attention_mode,
597+
model_sampling_override=config.get("model_sampling_override", "none"),
598+
model_sampling_shift=config.get("model_sampling_shift", 1.73),
599+
model_sampling_flux_max_shift=config.get("model_sampling_flux_max_shift", 1.15),
600+
model_sampling_flux_base_shift=config.get("model_sampling_flux_base_shift", 0.5),
601+
use_advanced_sampling=config.get("use_advanced_sampling", False),
602+
advanced_guider=config.get("advanced_guider", "cfg_guider"),
603+
advanced_scheduler=config.get("advanced_scheduler", "basic"),
604+
flux_guidance_value=config.get("flux_guidance_value", 0.0),
605+
width=w,
606+
height=h
597607
)
598608

599609
# --- VAE Decode ---

0 commit comments

Comments
 (0)