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

Merge 26f27d6f15 into d94738a762

Browse Source
This commit is contained in:
Sypherd 2024-09-25 20:22:08 +08:00 committed by GitHub
commit 43b4f12398
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
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