Skip to content

Unify Edge exporter run scripts - #4723

Open
micwill755 wants to merge 11 commits into
kimifrom
exporters-unified-runner
Open

micwill755 wants to merge 11 commits into
kimifrom
exporters-unified-runner

Conversation

@micwill755

Copy link
Copy Markdown
Collaborator

Summary

  • replace five model-specific smoke scripts with one run_export.py CLI
  • keep model loading and sample preparation in lazy family-specific prepare_export modules
  • mirror the Hugging Face exporter call shape: export(model, sample_inputs, config)
  • preserve per-family checkpoint, sequence-length, engine-directory, and output defaults

Validation

  • focused pre-commit hooks pass
  • unified CLI help smoke test passes
  • run_export.py kimi completes end to end with 100% eager/TRT close and relative L2 0.0005

Stack

This PR is stacked on the Kimi-K3 exporter PR.

Drop the Test/trt import path so PI05, GR00T, and Nemotron export is self-contained, and rename GR00T action_context to context_projection.
Each family now has spec, helpers, and patches in one folder, with shared vision/language/action-step code in models/common.
Compile the original HF/LeRobot submodules under apply_patches and stitch them with execute_engine, matching HF DynamoExporter.
…A export.

Keep Hub load stubs next to the Nemotron spec, and mention PI05/GR00T run_vla.py in the LLM tools README.
Document EdgeExporter.export, the returned ExportedProgram, family patches, and plugin converters, and link it from the user guide and HuggingFace tutorials.
@meta-cla meta-cla Bot added the cla signed label Sep 16, 2026
@micwill755 micwill755 changed the title Unify Edge exporter smoke runners Unify Edge exporter run scripts Sep 16, 2026
@micwill755
micwill755 added this pull request to stack #4727 September 21, 2026 21:58

This branch has not been deployed

No deployments
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant