| 266 | |
| 267 | |
| 268 | def parse_args(): |
| 269 | parser = argparse.ArgumentParser('', add_help=False) |
| 270 | parser.add_argument('--paths', help='+ separated list of checkpoints', required=True, |
| 271 | type=str) |
| 272 | parser.add_argument('--save_path', help='folder name to save optimized weights', default='optimized_logs', |
| 273 | type=str) |
| 274 | parser.add_argument('--categories', help='+ separated list of categories of the models', required=True, |
| 275 | type=str) |
| 276 | parser.add_argument('--prompts', help='prompts for composition model (can be a file or string)', default=None, |
| 277 | type=str) |
| 278 | parser.add_argument('--ckpt', required=True, |
| 279 | type=str) |
| 280 | parser.add_argument('--regularization_prompt', default='./data/regularization_captions.txt', |
| 281 | type=str) |
| 282 | return parser.parse_args() |
| 283 | |
| 284 | |
| 285 | if __name__ == "__main__": |