github/ai-toolkit
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Fix sampling issue on zimage turbo and krea2 turbo models

Browse Source
This commit is contained in:
Jaret Burkett
2026-07-08 12:34:39 -06:00
parent dd08579eda
commit 6233efe1bb
4 changed files with 12 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
extensions_built_in/diffusion_models/krea2/krea2.py
View File

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

6
extensions_built_in/diffusion_models/z_image/z_image.py
View File

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