Feature Extraction
Transformers
Safetensors
sentence-transformers
Chinese
English
qwen3_5
image-text-to-text
multimodal-embedding
text-embedding
image-embedding
video-embedding
mrl
custom_code
Instructions to use tencent/WeMM-Embedding-9B with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use tencent/WeMM-Embedding-9B with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("feature-extraction", model="tencent/WeMM-Embedding-9B", trust_remote_code=True)# Load model directly from transformers import AutoProcessor, AutoModelForMultimodalLM processor = AutoProcessor.from_pretrained("tencent/WeMM-Embedding-9B", trust_remote_code=True) model = AutoModelForMultimodalLM.from_pretrained("tencent/WeMM-Embedding-9B", trust_remote_code=True, device_map="auto") - sentence-transformers
How to use tencent/WeMM-Embedding-9B with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("tencent/WeMM-Embedding-9B", trust_remote_code=True) sentences = [ "The weather is lovely today.", "It's so sunny outside!", "He drove to the stadium." ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [3, 3] - Notebooks
- Google Colab
- Kaggle
Upload folder using huggingface_hub
Browse files- .gitattributes +0 -1
- README.md +5 -5
.gitattributes
CHANGED
|
@@ -1,2 +1 @@
|
|
| 1 |
*.safetensors filter=lfs diff=lfs merge=lfs -text
|
| 2 |
-
tokenizer.json filter=lfs diff=lfs merge=lfs -text
|
|
|
|
| 1 |
*.safetensors filter=lfs diff=lfs merge=lfs -text
|
|
|
README.md
CHANGED
|
@@ -30,10 +30,10 @@ import torch
|
|
| 30 |
from qwen_vl_utils import process_vision_info
|
| 31 |
from transformers import AutoModel, AutoProcessor
|
| 32 |
|
| 33 |
-
|
| 34 |
-
processor = AutoProcessor.from_pretrained(
|
| 35 |
model = AutoModel.from_pretrained(
|
| 36 |
-
|
| 37 |
).cuda().eval()
|
| 38 |
|
| 39 |
messages = [{"role": "user", "content": [
|
|
@@ -75,8 +75,8 @@ Use any subset of the content items to encode text, image, or video independentl
|
|
| 75 |
```python
|
| 76 |
from wemm_sentence_transformers import load_wemm_sentence_transformer
|
| 77 |
|
| 78 |
-
|
| 79 |
-
model = load_wemm_sentence_transformer(
|
| 80 |
inputs = [
|
| 81 |
"A dog is running on a beach.",
|
| 82 |
{"image": "/path/to/image.jpg", "text": "Represent this image."},
|
|
|
|
| 30 |
from qwen_vl_utils import process_vision_info
|
| 31 |
from transformers import AutoModel, AutoProcessor
|
| 32 |
|
| 33 |
+
model_id = "tencent/WeMM-Embedding-9B"
|
| 34 |
+
processor = AutoProcessor.from_pretrained(model_id, trust_remote_code=True)
|
| 35 |
model = AutoModel.from_pretrained(
|
| 36 |
+
model_id, trust_remote_code=True, dtype=torch.bfloat16
|
| 37 |
).cuda().eval()
|
| 38 |
|
| 39 |
messages = [{"role": "user", "content": [
|
|
|
|
| 75 |
```python
|
| 76 |
from wemm_sentence_transformers import load_wemm_sentence_transformer
|
| 77 |
|
| 78 |
+
model_id = "tencent/WeMM-Embedding-9B"
|
| 79 |
+
model = load_wemm_sentence_transformer(model_id, device="cuda:0")
|
| 80 |
inputs = [
|
| 81 |
"A dog is running on a beach.",
|
| 82 |
{"image": "/path/to/image.jpg", "text": "Represent this image."},
|