Update README.md

README.md

@@ -26,7 +26,9 @@ import requests
 from PIL import Image
 
 import torch
-from transformers import AutoProcessor, AutoModelForVision2Seq
+from transformers import AutoProcessor, AutoModelForVision2Seq, LlavaConfig
+import transformers
+from torch import nn
 
 class LlavaMultiModalProjectorYiVL(nn.Module):
     def __init__(self, config: "LlavaConfig"):
@@ -45,6 +47,7 @@ class LlavaMultiModalProjectorYiVL(nn.Module):
         hidden_states = self.linear_4(hidden_states)
         return hidden_states
 
+
 transformers.models.llava.modeling_llava.LlavaMultiModalProjector = LlavaMultiModalProjectorYiVL
 
 model_id = "BUAADreamer/Yi-VL-34B-hf"