diff --git a/src/diffusers/models/unet_2d.py b/src/diffusers/models/unet_2d.py index e4aea20d3a33..38e26422e2a7 100644 --- a/src/diffusers/models/unet_2d.py +++ b/src/diffusers/models/unet_2d.py @@ -293,7 +293,7 @@ def forward( emb = emb + class_emb elif self.class_embedding is None and class_labels is not None: raise ValueError("class_embedding needs to be initialized in order to use class conditioning") - + # 2. pre-process skip_sample = sample sample = self.conv_in(sample)