Skip to content

feat(export): support exporting sub-models from composite models#1031

Draft
xieofxie wants to merge 2 commits into
mainfrom
hualxie/export_submodule
Draft

feat(export): support exporting sub-models from composite models#1031
xieofxie wants to merge 2 commits into
mainfrom
hualxie/export_submodule

Conversation

@xieofxie

@xieofxie xieofxie commented Jul 3, 2026

Copy link
Copy Markdown
Contributor

Add --submodel option to winml export for selecting a specific sub-model from composite models (e.g., encoder, decoder).

When a composite model is detected:

  • --submodel : exports only that component as {stem}_{name}.onnx
  • No --submodel: exports all sub-models automatically
  • Non-composite models: unchanged single-model export

Extract _export_single_model() helper shared by both paths and add _resolve_composite_model_components() for registry/auto-detection.

Resolves #1014

hualxie added 2 commits July 2, 2026 17:34
Add --submodel option to `winml export` for selecting a specific
sub-model from composite models (e.g., encoder, decoder).

When a composite model is detected:
- --submodel <name>: exports only that component as {stem}_{name}.onnx
- No --submodel: exports all sub-models automatically
- Non-composite models: unchanged single-model export

Extract _export_single_model() helper shared by both paths and add
_resolve_composite_model_components() for registry/auto-detection.

Resolves #1014
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.

feat: Composite model: support export, build, perf etc. one or any of it sub models

1 participant