diff --git a/minigpt4/models/eva_vit.py b/minigpt4/models/eva_vit.py
index 7fcc63a..7807585 100644
--- a/minigpt4/models/eva_vit.py
+++ b/minigpt4/models/eva_vit.py
@@ -199,7 +199,7 @@ class PatchEmbed(nn.Module):
         B, C, H, W = x.shape
         # FIXME look at relaxing size constraints
         assert H == self.img_size[0] and W == self.img_size[1], \
-            f"Input image size ({H}*{W}) doesn't match model ({self.img_size[0]}*{self.img_size[1]})."
+            f"Input image size ({self.img_size[0]}*{self.img_size[1]}) doesn't match model ({H}*{W})."
         x = self.proj(x).flatten(2).transpose(1, 2)
         return x