Update README.md

README.md

@@ -49,7 +49,7 @@ pip install -U llama-index-embeddings-huggingface
 from llama_index.embeddings.huggingface import HuggingFaceEmbedding
 
 model = HuggingFaceEmbedding(
-    model_name_or_path="llamaindex/vdr-2b-v1",
+    model_name="llamaindex/vdr-2b-v1",
     device="mps",
     trust_remote_code=True,
 )
@@ -214,9 +214,9 @@ from sentence_transformers import SentenceTransformer
 
 model = SentenceTransformer(
     model_name_or_path="llamaindex/vdr-2b-v1",
-    device="mps",
+    device="cuda",
     trust_remote_code=True,
-    # These are the recommended kwargs for the model, but change them as needed
+    # These are the recommended kwargs for the model, but change them as needed if you don't have CUDA
     model_kwargs={
         "torch_dtype": torch.bfloat16, 
         "device_map": "cuda:0",