Skip to content

Support ONNX models without token types (e.g. BAAI/bge-m3) - #208

Merged
sanikolaev merged 3 commits into
masterfrom
fix/bge-m3-onnx-inputs
Sep 19, 2026
Merged

sanikolaev merged 3 commits into
masterfrom
fix/bge-m3-onnx-inputs

Conversation

@sanikolaev

@sanikolaev sanikolaev commented Sep 18, 2026

Copy link
Copy Markdown
Collaborator

- Resolve matching ONNX external-data sidecars, including multiple files
- Build tensors from configured ONNX inputs and select configured outputs
- Apply configured CLS or mean pooling before normalization
- Cover BGE-M3 embedding generation and output values
- Adjust RoBERTa-family lengths while preserving other model limits
- Extend tests for default and explicit padding offsets

@donhardman donhardman left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Applied required fixes to be clean implemenation

@sanikolaev
sanikolaev merged commit cebf86a into master Sep 19, 2026
10 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants