omriel1 commited on
Commit
9c96694
1 Parent(s): b4c270e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -2
README.md CHANGED
@@ -23,8 +23,7 @@ def get_device() -> str:
23
  return "mps"
24
  elif torch.cuda.is_available():
25
  return "cuda"
26
- else:
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(