def create_model()

in improved_diffusion/script_util.py [0:0]

• 12 lines of code
• 1 McCabe index (conditional complexity)

def create_model(
    image_size,
    num_channels,
    num_res_blocks,
    learn_sigma,
    class_cond,
    use_checkpoint,
    attention_resolutions,
    num_heads,
    num_heads_upsample,
    use_scale_shift_norm,
    dropout,