ssmits
/

Qwen2-7B-Instruct-embed-base

Text Classification

sentence-transformers

Model card Files Files and versions Community

ssmits commited on Jun 8

Commit

3876ca7

•

1 Parent(s): 0750bb8

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 = [