Skip to content

Conversation

@dg845
Copy link
Collaborator

@dg845 dg845 commented Nov 22, 2023

What does this PR do?

This PR fixes an issue in LCM Stable Diffusion distillation script /examples/consistency_distillation/train_lcm_distill_sd_wds.py related to the unet_time_cond_proj_dim parser argument. The script should accept and handle unet_time_cond_proj_dim gracefully after this change, whether the teacher checkpoint U-Net has time_cond_proj_dim set (e.g. a non null value) or not.

Fixes #5829.

Before submitting

Who can review?

Anyone in the community is free to review the PR once the tests have passed. Feel free to tag
members/contributors who may be interested in your PR.

@patrickvonplaten
@patil-suraj
@sayakpaul
@justindujardin

@dg845 dg845 changed the title Fix bug related to parsing unet_time_cond_proj_dim. Fix LCM Stable Diffusion distillation bug related to parsing unet_time_cond_proj_dim Nov 22, 2023
Copy link
Contributor

@patil-suraj patil-suraj left a comment

Choose a reason for hiding this comment

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

Good catch, thanks for fixing !

@patil-suraj patil-suraj merged commit 07eac4d into huggingface:main Nov 27, 2023
AmericanPresidentJimmyCarter pushed a commit to AmericanPresidentJimmyCarter/diffusers that referenced this pull request Apr 26, 2024
…e_cond_proj_dim (huggingface#5893)

* Fix bug related to parsing unet_time_cond_proj_dim.

* Fix analogous bug in the SD-XL LCM distillation script.
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.

LCM train scripts crash due to missing unet_time_cond_proj_dim argument

2 participants