Qwen3-Embedding-4B-rbln-tp2
Qwen/Qwen3-Embedding-4B ๋ฅผ Rebellions RBLN-CA25 NPU์ฉ์ผ๋ก ์ปดํ์ผํ ๋ชจ๋ธ.
- RBLN SDK: rebel-compiler 0.11.0 / optimum-rbln 0.11.0.post1
- ํ๊น: RBLN-CA25 (15.7 GiB/device)
- ์ปดํ์ผ config:
num_devices=2, max_seq_len=8192, batch_size=16, attn_impl=flash_attn, partition=4096 - โ ๏ธ ๋ฐํ์ device ๊ฐ์ = ์ปดํ์ผ num_devices โ ์ด ๋น๋๋ 2์ฅ ์ ์ฉ ([0,1]).
๋ก๋
from optimum.rbln import RBLNAutoModel
model = RBLNAutoModel.from_pretrained("Qwen3-Embedding-4B-rbln-tp2", export=False, rbln_config={"device": [0,1]})
์๋ฒ ๋ฉ: last-token ํ๋ง + ์ฟผ๋ฆฌ instruction ํ๋ฆฌํฝ์ค. vLLM --task embed๋ก๋ ์๋น ๊ฐ๋ฅ.
ํ ํฌ๋์ด์ /ํ๋ก์ธ์๊ฐ ์์ผ๋ฉด base ๋ชจ๋ธ(Qwen/Qwen3-Embedding-4B)์์ ๋ก๋ํ์ธ์.
- Downloads last month
- 8