diff --git a/comfy/ldm/hidream_o1/model.py b/comfy/ldm/hidream_o1/model.py index 6f8c77482..80352da59 100644 --- a/comfy/ldm/hidream_o1/model.py +++ b/comfy/ldm/hidream_o1/model.py @@ -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}"