ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-04-28 10:11:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixed issue with flux kontext forcing generation image sizes

Browse Source
This commit is contained in:
Jaret Burkett
2025-06-29 05:38:20 -06:00
parent 57a81bc0db
commit 5906a76666
3 changed files with 13 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
extensions_built_in/diffusion_models/flux_kontext/flux_kontext.py
View File

@@ -217,6 +217,13 @@ class FluxKontextModel(BaseModel):
else:
control_img = Image.open(gen_config.ctrl_img)
control_img = control_img.convert("RGB")
# resize to width and height
if control_img.size != (gen_config.width, gen_config.height):
control_img = control_img.resize(
(gen_config.width, gen_config.height), Image.BILINEAR
)
gen_config.width = int(gen_config.width // 16 * 16)
gen_config.height = int(gen_config.height // 16 * 16)
img = pipeline(
image=control_img,
prompt_embeds=conditional_embeds.text_embeds,
@@ -227,6 +234,8 @@ class FluxKontextModel(BaseModel):
guidance_scale=gen_config.guidance_scale,
latents=gen_config.latents,
generator=generator,
max_area=gen_config.height * gen_config.width,
_auto_resize=False,
**extra
).images[0]
return img