Fix sampling issue on zimage turbo and krea2 turbo models

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,

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,

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',

version.py

@@ -1 +1 @@
-VERSION = "0.10.20"
+VERSION = "0.10.21"