MCPcopy Create free account
hub / github.com/horseee/DeepCache / get_down_block

Function get_down_block

DeepCache/sd/unet_2d_blocks.py:38–244  ·  view source on GitHub ↗
(
    down_block_type,
    num_layers,
    in_channels,
    out_channels,
    temb_channels,
    add_downsample,
    resnet_eps,
    resnet_act_fn,
    transformer_layers_per_block=1,
    num_attention_heads=None,
    resnet_groups=None,
    cross_attention_dim=None,
    downsample_padding=None,
    dual_cross_attention=False,
    use_linear_projection=False,
    only_cross_attention=False,
    upcast_attention=False,
    resnet_time_scale_shift="default",
    attention_type="default",
    resnet_skip_time_act=False,
    resnet_out_scale_factor=1.0,
    cross_attention_norm=None,
    attention_head_dim=None,
    downsample_type=None,
    dropout=0.0,
)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 1

__init__Method · 0.70

Calls 11

DownBlock2DClass · 0.70
AttnDownBlock2DClass · 0.70
SkipDownBlock2DClass · 0.70
AttnSkipDownBlock2DClass · 0.70
DownEncoderBlock2DClass · 0.70
KDownBlock2DClass · 0.70

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…