model-integration
Extend vLLM-Omni with custom diffusion pipelines, model registrations, TTS adapter contracts, and focused maintainer tests without running broad model suites by default.
- License 许可
- Apache-2.0
- 最近更新
- 2026-09-03
- Stars
- ★ 351
- Forks
- 27
01 前言 Front Matter
- name
- model-integration
- license
- Apache 2.0
- metadata
- { "disco-role": "operating" }
- description
- Extend vLLM-Omni with custom diffusion pipelines, model registrations, TTS adapter contracts, and focused maintainer tests without running broad model suites by default.
- disable-model-invocation
- true
02 资源文件 4
- 参考文档references/custom-pipeline-and-model-registration.md6.6 KB
- 参考文档references/maintainer-workflows.md4.7 KB
- 参考文档references/troubleshooting.md4.6 KB
- 脚本scripts/check_tts_adapter_contract.py8.8 KB
03 SKILL.md 正文
Model integration
Use this sub-skill when the user is editing or extending vLLM-Omni itself: adding a model family, wiring a custom diffusion pipeline, updating a TTS adapter, or choosing focused maintainer tests before a larger GPU run.
Route here for
- Creating a custom diffusion pipeline using
diffusion_load_format,custom_pipeline_args,WorkerWrapperBase, orCustomPipelineWorkerExtension. - Adding or updating an omni AR, diffusion, TTS, action/world-model, or model-extra integration.
- Registering a
model_type, pipeline topology, default deploy config, endpoint restrictions, or serving adapter. - Reviewing OpenAI TTS adapter shape and optional dependency behavior.
- Choosing focused CPU parser/config/unit tests before expensive GPU/full-model examples.
Route elsewhere
- Local Python inference with existing models: use the sibling offline-inference sub-skill.
- HTTP serving payloads and server launch commands: use the sibling online-serving sub-skill.
- Deploy YAML overlays, connectors, placement, or memory planning: use the sibling stage-configuration sub-skill.
- Choosing an already supported model/recipe/backend: use the sibling model-recipes sub-skill.
Start here
-
Read custom-pipeline-and-model-registration.md for extension surfaces and registration flow.
-
Read maintainer-workflows.md before selecting tests or editing broad model code.
-
For TTS adapter edits, run or adapt the safe static checker:
python scripts/check_tts_adapter_contract.py --adapter-file path/to/adapter.pyThe checker parses source; it does not import the adapter, load a model, or call audio services.
-
Use troubleshooting.md when registrations are not discovered, a deploy YAML does not match a pipeline, optional dependencies are missing, or a model test is too expensive.
Safety rules
- Do CPU/static checks before GPU e2e tests. Full-model examples usually require model checkpoints, CUDA/accelerator hardware, large VRAM, and time.
- Keep custom-pipeline examples self-contained. Do not require future agents to open original examples before implementing a pipeline.
- Treat vendored/upstream model code as parity-sensitive; avoid style rewrites that make future upstream diffs hard.
- If a change touches endpoint contracts, update serving payload guidance in online-serving as well.
- If a change touches deploy topology or defaults, update stage-configuration guidance as well.
04 同仓库更多 Skills
passnet-triton-opt
Make a single PassNet Triton kernel fast AND numerically faithful: performance model (when a replacement can win at all), block/grid/warp tuning, autotune policy, launch overhead, and per-op numeric recipes to pass the dtype baseline tolerances. Use when a pass matches and is correct but speedup below expected, or when correctness fails by small numeric margins.
passnet-solve
END-TO-END playbook for solving one PassNet sample: analyze the computation graphs, decide the optimization strategy (what to fuse, what to replace, what to leave alone), drive the iteration loop, and maximize the sample score. This is the ENTRY skill — invoke it first for any PassNet optimization task; it tells you when to use passnet-pattern-fusion, passnet-triton-opt and passnet-feedback.
passnet-skill
PassNet GPU kernel optimization via compiler passes. Design and implement Triton-based optimization passes, create pass files under ./pass_dir/, self-evaluate with pass_evaluator, and iterate to maximize GPU speedup.
passnet-pattern-fusion
Author PassNet pass files: write patterns that actually MATCH the FX graph, pick fusion regions (which ops to absorb into one Triton kernel), and structure multi-pass submissions with the mandatory shared-dispatch architecture. Use when creating or fixing pattern()/replacement_args()/replacement_func() files, when a pass "failed to match", or when deciding how to fuse multiple kernels/ops.
passnet-orchestrate
Round/batch planning and decision support for PassNet graph optimization. Use for multi-sample triage, worker allocation, eval-budget policy, and post-eval keep/revert decisions. For solving one concrete sample end to end, use passnet-solve as the entry skill; consult this skill only when you need broader planning or a specific decision gate.
passnet-feedback
Fast, deterministic feedback for PassNet work: pre-flight pattern/match verification WITHOUT burning a GPU evaluation, per-node bottleneck analysis of a sample's graphs, and eval-log parsing into per-variant status + estimated score + failure classification. Use BEFORE every GPU evaluation (check_pattern), at the START of a sample (analyze_graph), and AFTER every evaluation (parse_eval_log).