wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-05-15 03:27:24 +08:00
Code Issues Actions 22 Packages Projects Releases Wiki Activity

Merge 2adb1df5c4 into 1f28908d6e

Browse Source
This commit is contained in:
DASVISHNU 2026-05-14 10:32:09 +00:00 committed by GitHub
commit 25de029a1c
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/ldm/hidream_o1/model.py
View File

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