mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-05-15 03:27:24 +08:00
Address review feedback for input dimension validation
This commit is contained in:
parent
cbcadb3e61
commit
fcfd235548
@ -149,7 +149,7 @@ class HiDreamO1Transformer(nn.Module):
|
||||
|
||||
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}"
|
||||
|
||||
Loading…
Reference in New Issue
Block a user