Skip to content

GitLab

"git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "73b125df68183279b90d21200c9a712b05e80083"
Unverified Commit 5058d27f authored by Charchit Sharma's avatar Charchit Sharma Committed by GitHub
Browse files

added attention_head_dim, attention_type, resolution_idx (#6011)

parent 748c1b3e
Hide whitespace changes
Inline Side-by-side
Showing with 7 additions and 0 deletions
src/diffusers/pipelines/versatile_diffusion/modeling_text_unet.py
View file @ 5058d27f
...@@ -50,6 +50,9 @@ def get_down_block( ...@@ -50,6 +50,9 @@ def get_down_block(
resnet_eps, resnet_eps,
resnet_act_fn, resnet_act_fn,
num_attention_heads, num_attention_heads,
transformer_layers_per_block,
attention_type,
attention_head_dim,
resnet_groups=None, resnet_groups=None,
cross_attention_dim=None, cross_attention_dim=None,
downsample_padding=None, downsample_padding=None,
...@@ -113,6 +116,10 @@ def get_up_block( ...@@ -113,6 +116,10 @@ def get_up_block(
resnet_eps, resnet_eps,
resnet_act_fn, resnet_act_fn,
num_attention_heads, num_attention_heads,
transformer_layers_per_block,
resolution_idx,
attention_type,
attention_head_dim,
resnet_groups=None, resnet_groups=None,
cross_attention_dim=None, cross_attention_dim=None,
dual_cross_attention=False, dual_cross_attention=False,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment