Fix validation variable name and formatting

This commit is contained in:
unknown 2026-05-14 16:01:51 +05:30
parent fcfd235548
commit 2adb1df5c4

View File

@ -154,6 +154,7 @@ class HiDreamO1Transformer(nn.Module):
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