From 971c9e3518f8d96ddff2355c77415ced68c63d22 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jukka=20Sepp=C3=A4nen?=
 <40791699+kijai@users.noreply.github.com>
Date: Mon, 18 May 2026 08:17:05 +0300
Subject: [PATCH] HiDream-O1: support area conditioning (#13944)

---
 comfy/model_base.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/comfy/model_base.py b/comfy/model_base.py
index 0736321b3..c22705655 100644
--- a/comfy/model_base.py
+++ b/comfy/model_base.py
@@ -1691,6 +1691,13 @@ class HiDreamO1(BaseModel):
         if text_input_ids is None or noise is None:
             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(
             text_input_ids, noise,
             ref_images=kwargs.get("reference_latents", None),