From 26f27d6f150f160051aaa06b8b682e83e3e05fbc Mon Sep 17 00:00:00 2001 From: Sypherd Date: Fri, 29 Sep 2023 23:49:51 -0600 Subject: [PATCH] Fix model and image size mismatch message --- minigpt4/models/eva_vit.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/minigpt4/models/eva_vit.py b/minigpt4/models/eva_vit.py index 7fcc63a7..78075852 100644 --- a/minigpt4/models/eva_vit.py +++ b/minigpt4/models/eva_vit.py @@ -199,7 +199,7 @@ def forward(self, x, **kwargs): 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