Address review feedback for input dimension validation

This commit is contained in:
unknown 2026-05-14 15:58:08 +05:30
parent cbcadb3e61
commit fcfd235548

View File

@ -149,7 +149,7 @@ 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}"