AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-29 07:22:12 +03:00
Code Activity

debugging

Browse Source
This commit is contained in:
sayakpaul
2023-08-23 16:13:34 +05:30
parent 6c5d3766d7
commit 99788b247f
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/diffusers/models/unet_2d_condition.py
View File

@@ -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)