()
| 111 | |
| 112 | |
| 113 | def create_argparser(): |
| 114 | defaults = dict( |
| 115 | clip_denoised=True, |
| 116 | num_samples=10000, |
| 117 | batch_size=16, |
| 118 | use_ddim=False, |
| 119 | model_path="", |
| 120 | classifier_path="", |
| 121 | classifier_scale=1.0, |
| 122 | ) |
| 123 | defaults.update(model_and_diffusion_defaults()) |
| 124 | defaults.update(classifier_defaults()) |
| 125 | parser = argparse.ArgumentParser() |
| 126 | add_dict_to_argparser(parser, defaults) |
| 127 | return parser |
| 128 | |
| 129 | |
| 130 | if __name__ == "__main__": |
no test coverage detected