Sentence Similarity
Transformers
Safetensors
sentence-transformers
Chinese
utu
feature-extraction
text-embeddings-inference
custom_code
Instructions to use tencent/Youtu-Embedding with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use tencent/Youtu-Embedding with Transformers:
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("tencent/Youtu-Embedding", trust_remote_code=True, dtype="auto") - sentence-transformers
How to use tencent/Youtu-Embedding with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("tencent/Youtu-Embedding", trust_remote_code=True) sentences = [ "那是 個快樂的人", "那是 條快樂的狗", "那是 個非常幸福的人", "今天是晴天" ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [4, 4] - Notebooks
- Google Colab
- Kaggle
Rename modeling_utu.py to modeling_youtu.py
Browse files
modeling_utu.py → modeling_youtu.py
RENAMED
|
@@ -53,7 +53,7 @@ from transformers.utils import (
|
|
| 53 |
replace_return_docstrings,
|
| 54 |
)
|
| 55 |
from transformers.utils.deprecation import deprecate_kwarg
|
| 56 |
-
from .
|
| 57 |
|
| 58 |
if is_torch_flex_attn_available():
|
| 59 |
from torch.nn.attention.flex_attention import BlockMask
|
|
|
|
| 53 |
replace_return_docstrings,
|
| 54 |
)
|
| 55 |
from transformers.utils.deprecation import deprecate_kwarg
|
| 56 |
+
from .configuration_youtu import UTUConfig
|
| 57 |
|
| 58 |
if is_torch_flex_attn_available():
|
| 59 |
from torch.nn.attention.flex_attention import BlockMask
|