Update README.md
Browse files
README.md
CHANGED
@@ -23,8 +23,7 @@ def get_device() -> str:
|
|
23 |
return "mps"
|
24 |
elif torch.cuda.is_available():
|
25 |
return "cuda"
|
26 |
-
|
27 |
-
return "cpu"
|
28 |
|
29 |
|
30 |
l2v = LLM2Vec.from_pretrained(
|
|
|
23 |
return "mps"
|
24 |
elif torch.cuda.is_available():
|
25 |
return "cuda"
|
26 |
+
return "cpu"
|
|
|
27 |
|
28 |
|
29 |
l2v = LLM2Vec.from_pretrained(
|