diffusers
0bee4d33 - LCM Distill Scripts Fix Bug when Initializing Target U-Net (#6848)

Commit
1 year ago
LCM Distill Scripts Fix Bug when Initializing Target U-Net (#6848) * Initialize target_unet from unet rather than teacher_unet so that we correctly add time_embedding.cond_proj if necessary. * Use UNet2DConditionModel.from_config to initialize target_unet from unet's config. --------- Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>
Author
Parents
Loading