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

Fix validation variable name and formatting

Browse Source
This commit is contained in:
unknown 2026-05-14 16:01:51 +05:30
parent fcfd235548
commit 2adb1df5c4
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -148,12 +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:
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}"
)
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