From 49f6b848805742cd3a59a7d121e84bb0572975da Mon Sep 17 00:00:00 2001 From: unknown <913556700@qq.com> Date: Thu, 18 May 2023 14:03:59 +0800 Subject: [PATCH] remove some inappropriate assertions --- imagebind/models/image_bind.py | 2 +- minigpt4/models/bind_gpt4.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/imagebind/models/image_bind.py b/imagebind/models/image_bind.py index 6c6431c..cbe262e 100644 --- a/imagebind/models/image_bind.py +++ b/imagebind/models/image_bind.py @@ -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) diff --git a/minigpt4/models/bind_gpt4.py b/minigpt4/models/bind_gpt4.py index 8a56b98..6234c73 100644 --- a/minigpt4/models/bind_gpt4.py +++ b/minigpt4/models/bind_gpt4.py @@ -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)