mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-05-15 11:37:32 +08:00
Fix validation variable name and formatting
This commit is contained in:
parent
fcfd235548
commit
2adb1df5c4
@ -149,11 +149,12 @@ class HiDreamO1Transformer(nn.Module):
|
|||||||
|
|
||||||
B, _, H, W = x.shape
|
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(
|
raise ValueError(
|
||||||
f"Input dimensions ({H},{W}) must be divisible"
|
f"Input dimensions ({H}, {W}) must be divisible "
|
||||||
f"by patch size{self.patch_size}"
|
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
|
||||||
|
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user