Skip to content

refactor(diffusion): migrate bridges to model builders#4556

Draft
yaoyu-33 wants to merge 1 commit into
yuya/refactor-bridge-model-buildersfrom
yuya/refactor-diffusion-model-builders
Draft

refactor(diffusion): migrate bridges to model builders#4556
yaoyu-33 wants to merge 1 commit into
yuya/refactor-bridge-model-buildersfrom
yuya/refactor-diffusion-model-builders

Conversation

@yaoyu-33

Copy link
Copy Markdown
Contributor

Summary

  • migrate Flux, LLaDA 1.5, Nemotron Labs Diffusion, and Wan bridges to serializable ModelConfig plus standalone ModelBuilder contracts
  • move official diffusion recipes off model-provider construction paths
  • preserve exact legacy transformer defaults and compatibility behavior, including Flux openai_gelu and Nemotron block sizing
  • extend registered-bridge and provider-free consumer audits to diffusion

Stack

Validation

  • uvx pre-commit run --all-files
  • two independent subagent review passes completed; all confirmed findings fixed
  • EOS targeted unit tests are running

Signed-off-by: yaoyu-33 <yaoyu.094@gmail.com>
@copy-pr-bot

copy-pr-bot Bot commented Jun 28, 2026

Copy link
Copy Markdown

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

@yaoyu-33

Copy link
Copy Markdown
Contributor Author

/ok to test afaeece

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.

1 participant