sap-ai-research
/

miCSE

Sentence Similarity

feature-extraction

text-embeddings-inference

Inference Endpoints

Model card Files Files and versions Community

TJKlein commited on Nov 23, 2022

Commit

e9ec569

·

1 Parent(s): 50e0acf

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -87,7 +87,7 @@ import umap.plot as umap_plot
 if torch.backends.mps.is_available():
     device = torch.device("mps")
 elif torch.cuda.is_available():
-    device = torch.device("gpu")
 else:
     device = torch.device("cpu")

 if torch.backends.mps.is_available():
     device = torch.device("mps")
 elif torch.cuda.is_available():
+    device = torch.device("cuda")
 else:
     device = torch.device("cpu")