-
dg845 authored
* 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>
0bee4d33