GitHub - alexrah/torch-export-splade-onnx: Torch wrapper to export SPLADE model with the entire pipeline including MLM (Masked Language Modeling) head

alexrah / torch-export-splade-onnx Public

Torch wrapper to export SPLADE model with the entire pipeline including MLM (Masked Language Modeling) head

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 2 Commits
.gitignore		.gitignore
.python-version		.python-version
README.md		README.md
main.py		main.py
pyproject.toml		pyproject.toml
uv.lock		uv.lock

Repository files navigation

Torch wrapper to export SPLADE model with the entire pipeline including MLM (Masked Language Modeling) head

No releases published

No packages published