From 99788b247ffdd5652f822dbff934928236fad7a4 Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Wed, 23 Aug 2023 16:13:34 +0530 Subject: [PATCH] debugging --- src/diffusers/models/unet_2d_condition.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/diffusers/models/unet_2d_condition.py b/src/diffusers/models/unet_2d_condition.py index 48a04bd9d1..f041bf56ae 100644 --- a/src/diffusers/models/unet_2d_condition.py +++ b/src/diffusers/models/unet_2d_condition.py @@ -920,7 +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)}") + print(f"From UNet before down blocks: {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 @@ -967,7 +967,7 @@ class UNet2DConditionModel(ModelMixin, ConfigMixin, UNet2DConditionLoadersMixin) encoder_attention_mask=encoder_attention_mask, ) # To support T2I-Adapter-XL - print(f"From UNet: {len(down_block_additional_residuals)}") + print(f"From UNet in mid block: {len(down_block_additional_residuals)}") if is_adapter and len(down_block_additional_residuals) > 0: sample += down_block_additional_residuals.pop(0)