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

Merge 2adb1df5c4 into 616cab4f97

Browse Source
This commit is contained in:
DASVISHNU 2026-05-15 00:37:38 +02:00 committed by GitHub
commit 49e2204b82
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