Add Muse Glimmer 30B CUDA INT4 recipe - #578
Justin Chu (justinchuby) wants to merge 11 commits into
Conversation
Export the native BF16 multimodal checkpoint with Mobius and quantize it to CUDA-targeted Q4 K-Quant INT4. Include reproducible dependencies, model metadata, documentation, and text/image ORT GenAI inference. Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com> Signed-off-by: justinchuby <justinchuby@users.noreply.github.com>
There was a problem hiding this comment.
Pull request overview
Adds a new Olive CUDA INT4 recipe folder for meta-models/Muse-Glimmer-30B, including build config, metadata, and a small streaming inference script to generate an ORT GenAI package via Mobius export + K-Quant.
Changes:
- Add CUDA INT4 Olive pipeline config (
MobiusBuilderBF16 export →OnnxKQuantQuantizationINT4) and output ignore rules. - Add end-user documentation (build/inference/validation) and an inference entrypoint using
onnxruntime_genai. - Register the recipe via
info.ymland add a recipe-localrequirements.txtincluding a pinned Mobius git dependency.
Reviewed changes
Copilot reviewed 7 out of 7 changed files in this pull request and generated 1 comment.
Show a summary per file
| File | Description |
|---|---|
| meta-models-Muse-Glimmer-30B/requirements.txt | Adds recipe dependencies, including a pinned Mobius git ref and Olive GPU extras. |
| meta-models-Muse-Glimmer-30B/README.md | Documents conversion pipeline, prerequisites, build, inference, and validation references. |
| meta-models-Muse-Glimmer-30B/LICENSE | Adds Apache 2.0 license text for the recipe/model distribution context. |
| meta-models-Muse-Glimmer-30B/info.yml | Registers the recipe for indexing (arch/device/EP + config path). |
| meta-models-Muse-Glimmer-30B/inference.py | Provides a minimal streaming inference script for the generated ORT GenAI package. |
| meta-models-Muse-Glimmer-30B/cuda/int4/config.json | Defines the Olive pipeline and CUDA target/output directory. |
| meta-models-Muse-Glimmer-30B/.gitignore | Ignores Olive cache and generated model artifacts. |
💡 Add a code-review agent skill or configure MCP servers for context-aware, tailored reviews. Learn more in the docs.
Keep the recipe reproducible as the Mobius repository grows by avoiding an ambiguous abbreviated object name. Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com> Signed-off-by: justinchuby <justinchuby@users.noreply.github.com>
|
CI note: the earlier |
Use the published mobius-onnx distribution name so the pinned Muse dependency installs successfully. Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com> Signed-off-by: justinchuby <justinchuby@users.noreply.github.com>
Pin the packed-image processor fix, include torchvision for processor export, document the ORT GenAI runtime dependency, and disable the incompatible cuDNN SDPA auto-path for long Hopper prefills. Record the measured full conversion and text/image CUDA smoke results. Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com> Signed-off-by: justinchuby <justinchuby@users.noreply.github.com>
|
Full H200 E2E is now complete. The exact recipe finished BF16 export + INT4 K-Quant in 1,436 seconds and produced an 18 GB three-model ORT GenAI package (decoder 15.4 GB, embedding 2.7 GB, vision 1.1 GB external data). Runtime smoke results with microsoft/onnxruntime-genai#2397:
The run exposed and fixed three integration gaps: the distribution name is |
Pin the Mobius revision that enables decoder CUDA Graph capture by default while keeping one-shot multimodal sessions uncaptured. Document the generated runtime behavior validated with text and image inference. Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com> Signed-off-by: justinchuby <justinchuby@users.noreply.github.com>
Pin the Mobius revision that emits fused ONNX RMSNormalization and residual SkipSimplifiedLayerNormalization operators instead of decomposed elementwise norm graphs. Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com> Signed-off-by: justinchuby <justinchuby@users.noreply.github.com>
Update the Mobius pin to preserve fp16 and bf16 output types for scale-free fused RMSNormalization nodes. Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com> Signed-off-by: justinchuby <justinchuby@users.noreply.github.com>
Use the Mobius revision that preserves Muse float32 RMSNorm semantics while emitting fused ONNX RMSNormalization operators, preventing BF16 generation drift. Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com> Signed-off-by: justinchuby <justinchuby@users.noreply.github.com>
Pin the recipe to the exact full SHA for the validated Muse RMSNorm implementation. Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com> Signed-off-by: justinchuby <justinchuby@users.noreply.github.com>
Use the Mobius revision that supplies schema-valid vector scales to ONNX RMSNormalization so the CUDA kernels produce correct output. Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com> Signed-off-by: justinchuby <justinchuby@users.noreply.github.com>
|
Final recipe validation for
|
Summary
Adds a CUDA recipe for
meta-models/Muse-Glimmer-30B:MobiusBuilder;OnnxKQuantQuantization(bits=4,block_size=32) for a CUDA-targeted package under 20 GB;mobius-onnxto the exact Muse implementation validated in Add Muse Glimmer 30B support onnxruntime/mobius#475 until that support is released.Validation
Native Muse loading currently requires microsoft/onnxruntime-genai#2397.
Co-authored-by: Copilot 223556219+Copilot@users.noreply.github.com