def create_argparser()

in scripts/ternary_search.py [0:0]

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

def create_argparser():
    defaults = dict(
        begin="0",
        end="39",
        training_mode="consistency_distillation",
        generator="determ",
        clip_denoised=True,
        num_samples=10000,
        batch_size=16,
        sampler="heun",
        s_churn=0.0,
        s_tmin=0.0,
        s_tmax=float("inf"),
        s_noise=1.0,
        steps=40,
        model_path="",
        ref_batch="",
        seed=42,
    )
    defaults.update(model_and_diffusion_defaults())
    parser = argparse.ArgumentParser()
    add_dict_to_argparser(parser, defaults)
    return parser