omriel1
/

LLM2Vec-DictaLM2.0-mntp-unsup-simcse

Model card Files Files and versions Community

omriel1 commited on 6 days ago

Commit

b4c270e

•

1 Parent(s): 9ad9da9

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -20,7 +20,7 @@ from llm2vec import LLM2Vec
 def get_device() -> str:
     if torch.backends.mps.is_available():
-        return "mps"  # mac GPU
     elif torch.cuda.is_available():
         return "cuda"
     else:

 def get_device() -> str:
     if torch.backends.mps.is_available():
+        return "mps"
     elif torch.cuda.is_available():
         return "cuda"
     else: