Views
No views yet

sentence-transformers>=4.1.0
transformers>=4.51.0
accelerate>=1.6.0
sentencepiece>=0.2.0
flash-attn>=2.7.31import torch
2from sentence_transformers import SentenceTransformer
3
4model_name = "iamtatsuki05/Sentence-Llama-Bi-JP-0.5B-PT"
5model_kwargs = {
6 "torch_dtype": torch.bfloat16,
7 "attn_implementation": "flash_attention_2",
8}
9model = SentenceTransformer(model_name, model_kwargs=model_kwargs)
10
11queries = ["ハチワレはどのようなキャラクターですか?"]
12docs = [
13 "ハチワレは、『ちいかわ』に登場する猫風のキャラクターで、明るく社交的、前向きな性格が特徴。ちいかわたちと共に日常を楽しみつつ、討伐などの冒険にも積極的に挑む存在です。",
14 "うさぎは、天真爛漫でマイペースな性格が特徴のキャラクターで、突飛な行動力と鋭い直感でちいかわたちを引っ張る存在。自由気ままながらも仲間思いな一面を併せ持ちます。",
15]
16q_emb = model.encode(queries, normalize_embeddings=True)
17d_emb = model.encode(docs, normalize_embeddings=True)
18scores = model.similarity(q_emb, d_emb)
19print(scores)| ID | Architecture | #Param. | #Param. w/o Emb. |
|---|---|---|---|
| iamtatsuki05/Sentence-ModernBERT-JP-0.5B-PT | ModernBERT | 679M | 548M |
| iamtatsuki05/Sentence-Llama-Bi-JP-0.5B-PT (this model) | Llama | 661M | 530M |
| iamtatsuki05/Sentence-Sarashina-Bi-0.5B-PT | Llama | 661M | 530M |
1@article{MIREI
2 title={同一条件下における Encoder/Decoder アーキテクチャによる文埋め込みの性能分析},
3 author={岡田 龍樹 and 杉本 徹},
4 journal={言語処理学会第 32 回年次大会 (NLP2026)},
5 year={2026}
6}