MCPcopy Create free account
hub / github.com/ImprintLab/MedSegDiff / create_model_and_diffusion

Function create_model_and_diffusion

guided_diffusion/script_util.py:77–136  ·  view source on GitHub ↗
(
    image_size,
    class_cond,
    learn_sigma,
    num_channels,
    num_res_blocks,
    channel_mult,
    in_ch,
    num_heads,
    num_head_channels,
    num_heads_upsample,
    attention_resolutions,
    dropout,
    diffusion_steps,
    noise_schedule,
    timestep_respacing,
    use_kl,
    predict_xstart,
    rescale_timesteps,
    rescale_learned_sigmas,
    use_checkpoint,
    use_scale_shift_norm,
    resblock_updown,
    use_fp16,
    use_new_attention_order,
    dpm_solver,
    version,
)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 2

mainFunction · 0.90
mainFunction · 0.90

Calls 2

create_modelFunction · 0.85

Tested by

no test coverage detected