Hugging Face's logo

ssmits
/
Qwen2-7B-Instruct-embed-base

Text Classification
sentence-transformers
Safetensors
English
qwen2
pretrained
Model card Files
xet
Community
1
ssmits commited on
Commit
3876ca7
·
verified ·
1 Parent(s): 0750bb8

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -28,7 +28,7 @@ from sentence_transformers import SentenceTransformer
28
  import torch
29
 
30
  # 1. Load a pretrained Sentence Transformer model
31
- model = SentenceTransformer("ssmits/Qwen2-7B-embed-base", device = "cpu")
32
 
33
  # The sentences to encode
34
  sentences = [
 
28
  import torch
29
 
30
  # 1. Load a pretrained Sentence Transformer model
31
+ model = SentenceTransformer("ssmits/Qwen2-7B-embed-base") # device = "cpu" when <= 24 GB VRAM
32
 
33
  # The sentences to encode
34
  sentences = [