From d44c0b01825d199235851ef425c706b5ce415435 Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Wed, 23 Aug 2023 15:58:51 +0530 Subject: [PATCH] debugging --- src/diffusers/models/unet_2d_condition.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/diffusers/models/unet_2d_condition.py b/src/diffusers/models/unet_2d_condition.py index c2bf73a2fc..48a04bd9d1 100644 --- a/src/diffusers/models/unet_2d_condition.py +++ b/src/diffusers/models/unet_2d_condition.py @@ -920,6 +920,7 @@ class UNet2DConditionModel(ModelMixin, ConfigMixin, UNet2DConditionLoadersMixin) is_adapter = mid_block_additional_residual is None and down_block_additional_residuals is not None down_block_res_samples = (sample,) + print(f"From UNet: {len(down_block_additional_residuals)}") for downsample_block in self.down_blocks: if hasattr(downsample_block, "has_cross_attention") and downsample_block.has_cross_attention: # For t2i-adapter CrossAttnDownBlock2D