Update README.md

README.md

@@ -49,14 +49,14 @@ See https://llava-vl.github.io/ for more details.
 usage is as follows 
 
 ```python
-from transformers import LlavaProcessor, LlavaLlamaForCausalLM
+from transformers import LlavaProcessor, LlavaForCausalLM
 from PIL import Image
 import requests
 import torch
 
 PATH_TO_CONVERTED_WEIGHTS = "shauray/Llava-Llama-2-7B-hf"
 
-model = LlavaLlamaForCausalLM.from_pretrained(PATH_TO_CONVERTED_WEIGHTS,
+model = LlavaForCausalLM.from_pretrained(PATH_TO_CONVERTED_WEIGHTS,
 device_map="cuda",torch_dtype=torch.float16).to("cuda")
 processor = LlavaProcessor.from_pretrained(PATH_TO_CONVERTED_WEIGHTS)