wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-05-24 07:57:29 +08:00
Code Issues Actions 16 Packages Projects Releases Wiki Activity

HiDream-O1: support area conditioning (#13944)

Browse Source
This commit is contained in:
Jukka Seppänen 2026-05-18 08:17:05 +03:00 committed by GitHub
parent b39af210d0
commit 971c9e3518
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
comfy/model_base.py
View File

@ -1691,6 +1691,13 @@ class HiDreamO1(BaseModel):
if text_input_ids is None or noise is None: if text_input_ids is None or noise is None:
return out return out
# handle area conds
area = kwargs.get("area", None)
if area is not None:
crop_h = min(noise.shape[-2] - area[2], area[0])
crop_w = min(noise.shape[-1] - area[3], area[1])
noise = torch.empty((noise.shape[0], 3, crop_h, crop_w), dtype=noise.dtype, device=noise.device)
conds = build_extra_conds( conds = build_extra_conds(
text_input_ids, noise, text_input_ids, noise,
ref_images=kwargs.get("reference_latents", None), ref_images=kwargs.get("reference_latents", None),