Merge 2adb1df5c4 into 3fe9f5fecb

Fix validation variable name and formatting
Address review feedback for input dimension validation
2026-07-07 07:01:46 +08:00 · 2026-07-04 14:10:46 +08:00 · 2026-05-14 16:01:51 +05:30 · 2026-05-14 15:58:08 +05:30 · 2026-05-14 13:36:08 +05:30
1 changed files with 7 additions and 0 deletions
--- a/comfy/ldm/hidream_o1/model.py
+++ b/comfy/ldm/hidream_o1/model.py
@ -148,6 +148,13 @@ class HiDreamO1Transformer(nn.Module):
            raise ValueError("HiDreamO1Transformer requires input_ids and position_ids in conditioning")

        B, _, H, W = x.shape
+        
+        if H % self.patch_size != 0 or W % self.patch_size != 0:
+            raise ValueError(
+        f"Input dimensions ({H}, {W}) must be divisible "
+        f"by patch size {self.patch_size}"
+    )
+
        h_p, w_p = H // self.patch_size, W // self.patch_size
        tgt_image_len = h_p * w_p
Author	SHA1	Message	Date
DASVISHNU	5f7b28579d	Merge `2adb1df5c4` into `3fe9f5fecb`	2026-07-04 14:10:46 +08:00
unknown	2adb1df5c4	Fix validation variable name and formatting	2026-05-14 16:01:51 +05:30
unknown	fcfd235548	Address review feedback for input dimension validation	2026-05-14 15:58:08 +05:30
unknown	cbcadb3e61	Add validation for invalid HiDream input dimensions	2026-05-14 13:36:08 +05:30