Update README.md

README.md

@@ -30,6 +30,7 @@ from transformers import AutoProcessor, AutoModelForVision2Seq, LlavaConfig
 import transformers
 from torch import nn
 
+
 class LlavaMultiModalProjectorYiVL(nn.Module):
     def __init__(self, config: "LlavaConfig"):
         super().__init__()
@@ -46,8 +47,7 @@ class LlavaMultiModalProjectorYiVL(nn.Module):
         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-34B-hf"