| 139 | |
| 140 | |
| 141 | def parse_args(): |
| 142 | parser = argparse.ArgumentParser('Checkpoint conversion given delta ckpts, currently supported for stable diffusion 1.4 only', add_help=True) |
| 143 | parser.add_argument('--ckpt', help='pretrained compvis model checkpoint', required=True, |
| 144 | type=str) |
| 145 | parser.add_argument('--delta_ckpt', help='delta checkpoint either of compvis or diffuser', required=True, |
| 146 | type=str) |
| 147 | parser.add_argument('--sd_version', default="CompVis/stable-diffusion-v1-4", |
| 148 | type=str) |
| 149 | parser.add_argument('--config', default="configs/custom-diffusion/finetune.yaml", |
| 150 | type=str) |
| 151 | parser.add_argument('--modelname', default="model.ckpt", help="name of the model to save when converting to webui", |
| 152 | type=str) |
| 153 | parser.add_argument("--mode", default='compvis-to-diffuser', choices=['diffuser-to-webui', 'compvis-to-webui', 'compvis-to-diffuser', 'diffuser-to-compvis'], |
| 154 | type=str) |
| 155 | return parser.parse_args() |
| 156 | |
| 157 | |
| 158 | if __name__ == "__main__": |