ssmits
/

Qwen2-7B-embed-base

Text Classification

sentence-transformers

Model card Files Files and versions Community

ssmits commited on Jun 9

Commit

0f958e1

•

1 Parent(s): f459b34

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -28,7 +28,7 @@ from sentence_transformers import SentenceTransformer
 import torch
 # 1. Load a pretrained Sentence Transformer model
-model = SentenceTransformer("ssmits/Qwen2-7B-embed-base", device = "cpu")
 # The sentences to encode
 sentences = [

 import torch
 # 1. Load a pretrained Sentence Transformer model
+model = SentenceTransformer("ssmits/Qwen2-7B-embed-base") # device = "cpu" when <= 24 GB VRAM
 # The sentences to encode
 sentences = [