Skip to content

Add from_single_file() support for Ideogram4 and Krea2 transformers#14126

Open
buffett0323 wants to merge 4 commits into
huggingface:mainfrom
buffett0323:fix/single-file-ideogram4-krea2
Open

Add from_single_file() support for Ideogram4 and Krea2 transformers#14126
buffett0323 wants to merge 4 commits into
huggingface:mainfrom
buffett0323:fix/single-file-ideogram4-krea2

Conversation

@buffett0323

Copy link
Copy Markdown

What does this PR do?

Adds from_single_file() support for Ideogram4Transformer2DModel and
Krea2Transformer2DModel.

Fixes #14122

Before submitting

Who can review?

@sayakpaul @yiyixuxu @DN6

@github-actions github-actions Bot added documentation Improvements or additions to documentation models tests single-file fixes-issue size/L PR with diff > 200 LOC labels Jul 6, 2026
@sayakpaul sayakpaul requested a review from DN6 July 6, 2026 02:42
checkpoint[k.replace("model.diffusion_model.", "")] = checkpoint.pop(k)

return checkpoint

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Could we add a check to infer the model type automatically

def infer_diffusers_model_type(checkpoint):

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

Labels

documentation Improvements or additions to documentation fixes-issue models single-file size/L PR with diff > 200 LOC tests

Projects

None yet

Development

Successfully merging this pull request may close these issues.

from_single_file() support missing/incomplete for Ideogram4 and Krea2 transformers

2 participants