Add EximiusLabs/fusion-embedding-2-2b-preview (multimodal embedding: text, image, video, audio) - #741
Add EximiusLabs/fusion-embedding-2-2b-preview (multimodal embedding: text, image, video, audio)#741abtonmoy wants to merge 2 commits into
Conversation
|
The latest updates on your projects. Learn more about Vercel for GitHub.
|
There was a problem hiding this comment.
Code Review
This pull request introduces a configuration file and guide for the fusion-embedding-2-2b-preview multimodal embedding model. The feedback recommends replacing uv pip install with standard pip install for broader compatibility, increasing the minimum VRAM requirement to 8 GB to match standard hardware tiers, and wrapping the offline API input in a list to prevent potential type validation errors in vLLM.
| min_vllm_version: "0.26.0" | ||
| install: | ||
| pip: | ||
| command: "uv pip install 'vllm[audio]==0.26.0' fusion-embedding" |
| variants: | ||
| default: | ||
| precision: bf16 | ||
| vram_minimum_gb: 7 |
There was a problem hiding this comment.
Increasing vram_minimum_gb to 8 aligns with standard hardware tiers (such as 8 GB GPUs) and the guide's recommendation of 'roughly 8 GB VRAM or more'. A 7 GB limit is not a standard GPU size and is extremely tight for loading ~5.7 GB of weights plus the CUDA context and vLLM overhead.
vram_minimum_gb: 8| ## Install | ||
|
|
||
| ```bash | ||
| uv pip install 'vllm[audio]==0.26.0' fusion-embedding |
| out = llm.embed({"prompt": "<|vision_pad|><|im_end|>", | ||
| "multi_modal_data": {"audio": (wav, sr)}}) |
There was a problem hiding this comment.
|
Addressed the review: bumped |
Signed-off-by: Abdul Basit Tonmoy <abdulbasittonmoy@gmail.com>
Signed-off-by: Abdul Basit Tonmoy <abdulbasittonmoy@gmail.com>
3e50af9 to
5f8d48f
Compare
Adds a recipe for EximiusLabs/fusion-embedding-2-2b-preview, a 2.8B multimodal embedding model that places text, images, video and audio in one 2048-d space (frozen Qwen3-VL-Embedding-2B base + frozen Qwen2.5-Omni audio tower + token-gated adapters).
The model serves through an out-of-tree plugin registered via the
vllm.general_pluginsentry point; it ships inside thefusion-embeddingPyPI package, so the install isuv pip install 'vllm[audio]==0.26.0' fusion-embeddingand the launch is a plainvllm serve EximiusLabs/fusion-embedding-2-2b-preview --runner pooling(no trust-remote-code, no extra flags). Smoke-tested against vllm==0.26.0.Notes for review:
node scripts/build-recipes-api.mjspasses and renderspublic/EximiusLabs/fusion-embedding-2-2b-preview.json.single_node_tpwithtp: 1, eager execution forced by the plugin (documented in the guide).Happy to adjust tags, difficulty, or guide structure to fit conventions.