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

Add validation for invalid HiDream input dimensions

Browse Source
This commit is contained in:
unknown 2026-05-14 13:36:08 +05:30
parent 1f28908d6e
commit cbcadb3e61
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
comfy/ldm/hidream_o1/model.py
View File

@ -148,6 +148,12 @@ class HiDreamO1Transformer(nn.Module):
raise ValueError("HiDreamO1Transformer requires input_ids and position_ids in conditioning") raise ValueError("HiDreamO1Transformer requires input_ids and position_ids in conditioning")
B, _, H, W = x.shape 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 h_p, w_p = H // self.patch_size, W // self.patch_size
tgt_image_len = h_p * w_p tgt_image_len = h_p * w_p