ssmits commited on
Commit
b703267
1 Parent(s): 908ba85

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -33,7 +33,7 @@ from sentence_transformers import SentenceTransformer
33
  import torch
34
 
35
  # 1. Load a pretrained Sentence Transformer model
36
- model = SentenceTransformer("ssmits/Falcon2-5.5B-multilingual-embed-base")
37
 
38
  # The sentences to encode
39
  sentences = [
 
33
  import torch
34
 
35
  # 1. Load a pretrained Sentence Transformer model
36
+ model = SentenceTransformer("ssmits/Falcon2-5.5B-multilingual-embed-base") # device = "cpu" when <= 24 GB VRAM
37
 
38
  # The sentences to encode
39
  sentences = [