added attention_head_dim, attention_type, resolution_idx (#6011)

5058d27f · Charchit Sharma · GitHub · 748c1b3e · 5058d27f
Unverified Commit 5058d27f authored Dec 01, 2023 by Charchit Sharma Committed by GitHub Dec 01, 2023
Show whitespace changes
Inline Side-by-side

Showing with 7 additions and 0 deletions

src/diffusers/pipelines/versatile_diffusion/modeling_text_unet.py ...users/pipelines/versatile_diffusion/modeling_text_unet.py +7 -0

No files found.
--- a/src/diffusers/pipelines/versatile_diffusion/modeling_text_unet.py
+++ b/src/diffusers/pipelines/versatile_diffusion/modeling_text_unet.py
@@ -50,6 +50,9 @@ def get_down_block(
    resnet_eps,
    resnet_act_fn,
    num_attention_heads,
+    transformer_layers_per_block,
+    attention_type,
+    attention_head_dim,
    resnet_groups=None,
    cross_attention_dim=None,
    downsample_padding=None,
@@ -113,6 +116,10 @@ def get_up_block(
    resnet_eps,
    resnet_act_fn,
    num_attention_heads,
+    transformer_layers_per_block,
+    resolution_idx,
+    attention_type,
+    attention_head_dim,
    resnet_groups=None,
    cross_attention_dim=None,
    dual_cross_attention=False,