llamaindex
/

vdr-2b-v1

Image-Text-to-Text

sentence-transformers

Model card Files Files and versions Community

cheesyFishes commited on 8 days ago

Commit

601588d

·

verified ·

1 Parent(s): cfef054

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -50,7 +50,7 @@ from llama_index.embeddings.huggingface import HuggingFaceEmbedding
 model = HuggingFaceEmbedding(
     model_name="llamaindex/vdr-2b-v1",
-    device="mps",
     trust_remote_code=True,
 )

 model = HuggingFaceEmbedding(
     model_name="llamaindex/vdr-2b-v1",
+    device="cpu",  # "mps" for mac, "cuda" for nvidia GPUs
     trust_remote_code=True,
 )