diff --git a/extensions_built_in/diffusion_models/krea2/krea2.py b/extensions_built_in/diffusion_models/krea2/krea2.py index 75b6ae5..1440386 100644 --- a/extensions_built_in/diffusion_models/krea2/krea2.py +++ b/extensions_built_in/diffusion_models/krea2/krea2.py @@ -494,6 +494,9 @@ class Krea2Model(BaseModel): ref_latents = [ self._encode_ref_latents(ctrl_tensors, target_pixels=target_pixels) ] + + # CFG is 0 normalized for this model + guidance = max(0.0, gen_config.guidance_scale - 1.0) img = pipeline( conditional_embeds=conditional_embeds, @@ -501,7 +504,7 @@ class Krea2Model(BaseModel): height=gen_config.height, width=gen_config.width, num_inference_steps=gen_config.num_inference_steps, - guidance_scale=gen_config.guidance_scale, + guidance_scale=guidance, latents=gen_config.latents, generator=generator, ref_latents=ref_latents, diff --git a/extensions_built_in/diffusion_models/z_image/z_image.py b/extensions_built_in/diffusion_models/z_image/z_image.py index 17c7e07..6f7ba2c 100644 --- a/extensions_built_in/diffusion_models/z_image/z_image.py +++ b/extensions_built_in/diffusion_models/z_image/z_image.py @@ -423,13 +423,17 @@ class ZImageModel(BaseModel): sc = self.get_bucket_divisibility() gen_config.width = int(gen_config.width // sc * sc) gen_config.height = int(gen_config.height // sc * sc) + + # CFG is 0 normalized for this model + guidance = max(0.0, gen_config.guidance_scale - 1.0) + img = pipeline( prompt_embeds=conditional_embeds.text_embeds, negative_prompt_embeds=unconditional_embeds.text_embeds, height=gen_config.height, width=gen_config.width, num_inference_steps=gen_config.num_inference_steps, - guidance_scale=gen_config.guidance_scale, + guidance_scale=guidance, latents=gen_config.latents, generator=generator, **extra, diff --git a/ui/src/app/jobs/new/options.ts b/ui/src/app/jobs/new/options.ts index 7f52db0..2ef609b 100644 --- a/ui/src/app/jobs/new/options.ts +++ b/ui/src/app/jobs/new/options.ts @@ -621,7 +621,7 @@ export const modelArchs: ModelArch[] = [ undefined, ], 'config.process[0].sample.guidance_scale': [1, 4], - 'config.process[0].sample.sample_steps': [8, 25], + 'config.process[0].sample.sample_steps': [9, 25], }, disableSections: ['network.conv'], additionalSections: ['model.low_vram', 'model.layer_offloading', 'model.assistant_lora_path'], @@ -1077,7 +1077,7 @@ export const modelArchs: ModelArch[] = [ undefined, ], 'config.process[0].sample.guidance_scale': [1, 4], - 'config.process[0].sample.sample_steps': [8, 25], + 'config.process[0].sample.sample_steps': [9, 25], }, disableSections: [ 'network.conv', diff --git a/version.py b/version.py index dba99a5..01668c3 100644 --- a/version.py +++ b/version.py @@ -1 +1 @@ -VERSION = "0.10.20" +VERSION = "0.10.21"