BUAADreamer
/

Yi-VL-6B-hf

Visual Question Answering

image-text-to-text

Inference Endpoints

Model card Files Files and versions Community

BUAADreamer commited on May 21, 2024

Commit

0e71cb1

·

verified ·

1 Parent(s): a94341a

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -46,7 +46,7 @@ class LlavaMultiModalProjectorYiVL(nn.Module):
         hidden_states = self.linear_3(hidden_states)
         hidden_states = self.linear_4(hidden_states)
         return hidden_states
 transformers.models.llava.modeling_llava.LlavaMultiModalProjector = LlavaMultiModalProjectorYiVL
 model_id = "BUAADreamer/Yi-VL-6B-hf"

         hidden_states = self.linear_3(hidden_states)
         hidden_states = self.linear_4(hidden_states)
         return hidden_states
+# Monkey patch of LlavaMultiModalProjector is mandatory
 transformers.models.llava.modeling_llava.LlavaMultiModalProjector = LlavaMultiModalProjectorYiVL
 model_id = "BUAADreamer/Yi-VL-6B-hf"