-
Couldn't load subscription status.
- Fork 6.4k
[modular]some small fix #12307
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[modular]some small fix #12307
Conversation
|
The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update. |
At the moment relies on huggingface/diffusers#12307
|
|
||
|
|
||
| @dataclass(frozen=True) | ||
| class MellonParam: |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
MellonParam: this is to support UI element configuration in Mellon
we will support a direct conversion from InputParam
| class InputParam: |
| SUPPORTED_NODE_TYPES = {"controlnet", "vae_encoder", "denoise", "text_encoder", "decoder"} | ||
|
|
||
|
|
||
| # Mellon Input Parameters (runtime parameters, not models) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
we keep a common configuration for mellon inputs/model_onputs/outputs
| ] | ||
| ) | ||
|
|
||
| MODULAR_PIPELINE_BLOCKS_MAPPING = OrderedDict( |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
remove this map, each ModularPipeline should specify a default_blocks_name instead
| ) | ||
|
|
||
|
|
||
| class QwenImageCoreDenoiseStep(SequentialPipelineBlocks): |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
adding a *CoreDenoiseStep -> corresponding the "Denoise" node in mellon: bascially takes all the encoder inputs and output latents
No description provided.