Update README.md

README.md

@@ -52,7 +52,7 @@ pip install -U llama-index-embeddings-huggingface
 from llama_index.embeddings.huggingface import HuggingFaceEmbedding
 
 model = HuggingFaceEmbedding(
-    model_name="llamaindex/vdr-2b-v1",
+    model_name="llamaindex/vdr-2b-multi-v1",
     device="cpu",  # "mps" for mac, "cuda" for nvidia GPUs
     trust_remote_code=True,
 )
@@ -81,7 +81,7 @@ min_pixels = 1 * 28 * 28
 
 # Load the embedding model and processor
 model = Qwen2VLForConditionalGeneration.from_pretrained(
-    'llamaindex/vdr-2b-v1',
+    'llamaindex/vdr-2b-multi-v1',
     # These are the recommended kwargs for the model, but change them as needed
     attn_implementation="flash_attention_2",
     torch_dtype=torch.bfloat16,
@@ -89,7 +89,7 @@ model = Qwen2VLForConditionalGeneration.from_pretrained(
 ).eval()
 
 processor = AutoProcessor.from_pretrained(
-    'llamaindex/vdr-2b-v1',
+    'llamaindex/vdr-2b-multi-v1',
     min_pixels=min_pixels,
     max_pixels=max_pixels
 )
@@ -217,7 +217,7 @@ via SentenceTransformers
 from sentence_transformers import SentenceTransformer
 
 model = SentenceTransformer(
-    model_name_or_path="llamaindex/vdr-2b-v1",
+    model_name_or_path="llamaindex/vdr-2b-multi-v1",
     device="cuda",
     trust_remote_code=True,
     # These are the recommended kwargs for the model, but change them as needed if you don't have CUDA