MCPcopy Create free account
hub / github.com/NVlabs/DiffPure / create_model

Function create_model

guided_diffusion/script_util.py:138–192  ·  view source on GitHub ↗
(
    image_size,
    num_channels,
    num_res_blocks,
    channel_mult="",
    learn_sigma=False,
    class_cond=False,
    use_checkpoint=False,
    attention_resolutions="16",
    num_heads=1,
    num_head_channels=-1,
    num_heads_upsample=-1,
    use_scale_shift_norm=False,
    dropout=0,
    resblock_updown=False,
    use_fp16=False,
    use_new_attention_order=False,
)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 1

Calls 1

UNetModelClass · 0.85

Tested by

no test coverage detected