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

remove some inappropriate assertions

Browse Source
This commit is contained in:
unknown 2023-05-18 14:03:59 +08:00
parent d2db151df2
commit 49f6b84880
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
imagebind/models/image_bind.py
View File

@ -96,7 +96,7 @@ class ImageBindJoiner(nn.Module):
def forward(self, inputs: Dict[str, Tensor]) -> Dict[str, Tensor]:
outputs = {}
for modality_key, modality_value in inputs.items():
assert modality_key == ModalityType.VISION, "Only Vision is Currently Supported."
# assert modality_key == ModalityType.VISION, "Only Vision is Currently Supported."
if modality_value is not None:
modality_value = self.modality_pre_projectors[modality_key](modality_value)
modality_value = self.modality_qformers[modality_key](modality_value)

2
minigpt4/models/bind_gpt4.py
View File

@ -110,7 +110,7 @@ class BindGPT4(BaseModel):
Other modalities will conflict with the pre-defined prompt and wrapping strategy.
"""
embeds = self.encode_inputs(inputs)
assert "vision" in embeds, "Only Vision Input Can Be Accepted Now."
# assert "vision" in embeds, "Only Vision Input Can Be Accepted Now."
prompt = random.choice(self.prompt_list)
img_embeds, atts_img = self.prompt_wrap(embeds, ModalityType.VISION, prompt)