chris/MiniGPT-4
1
0
Fork 0
mirror of https://github.com/Vision-CAIR/MiniGPT-4.git synced 2025-04-05 02:20:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix model and image size mismatch message

Browse Source
This commit is contained in:
Sypherd 2023-09-29 23:49:51 -06:00
parent ef1ac08ce3
commit 26f27d6f15
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
minigpt4/models/eva_vit.py
View File

@ -199,7 +199,7 @@ class PatchEmbed(nn.Module):
B, C, H, W = x.shape B, C, H, W = x.shape
# FIXME look at relaxing size constraints # FIXME look at relaxing size constraints
assert H == self.img_size[0] and W == self.img_size[1], \ 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) x = self.proj(x).flatten(2).transpose(1, 2)
return x return x