Add vLLM and Sentence Transformers support for embedding generation - #1346

Merged

praateekmahajan merged 22 commits into

NVIDIA-NeMo:mainfrom

praateekmahajan:praateek/vllm-embeddings

Jan 14, 2026

Commits on Dec 31, 2025

..
praateekmahajan
committed
c06c705

Commits on Jan 5, 2026

failing tests
praateekmahajan
committed
d77ef3c
pin sklearn to < 1.8
praateekmahajan
committed
7a318d3

Commits on Jan 6, 2026

Merge branch 'main' into praateek/vllm-embeddings
ayushdg
authored
2894bdb

Commits on Jan 7, 2026

Commits on Jan 8, 2026

Merge branch 'main' of github.com:NVIDIA/NeMo-Curator into praateek/vllm-embeddings
praateekmahajan
committed
03f9399

Commits on Jan 12, 2026

pr reviews
praateekmahajan
committed
ec63c6f
trying ray 2.52
praateekmahajan
committed
7a9a282
f401 import vllm in test
praateekmahajan
committed
4e3303c
Merge branch 'main' of github.com:NVIDIA/NeMo-Curator into praateek/vllm-embeddings
praateekmahajan
committed
65c7c9c
move sentence transformers to text_cpi
praateekmahajan
committed
559f568
move imports around for gpu safety
praateekmahajan
committed
eda934a
move fixutre inside
praateekmahajan
committed
4c47bfc
double quote st
praateekmahajan
committed
a1fe5c3
move fixture out
praateekmahajan
committed
9eb824c
greptile comments
praateekmahajan
committed
8a67c40

Commits on Jan 13, 2026

have teardown module in qwen_lm
praateekmahajan
committed
11d0ebe
Merge branch 'main' into praateek/vllm-embeddings
sarahyurick
authored
bcc51ad
remove mock
praateekmahajan
committed
069e1ba
Merge branch 'praateek/vllm-embeddings' of github.com:praateekmahajan/NeMo-Curator into praateek/vllm-embeddings
praateekmahajan
committed
e0a6967
update to ray 2.52.1
praateekmahajan
committed
d4b99ee