| 249 | |
| 250 | |
| 251 | def get_parser(): |
| 252 | parser = argparse.ArgumentParser() |
| 253 | parser.add_argument("--savedir", type=str, default=None, help="results saving path") |
| 254 | parser.add_argument("--ckpt_path", type=str, default=None, help="checkpoint path") |
| 255 | parser.add_argument("--base", type=str, help="config (yaml) path") |
| 256 | parser.add_argument("--prompt_file", type=str, default=None, help="a text file containing many prompts") |
| 257 | parser.add_argument("--n_samples", type=int, default=1, help="num of samples per prompt",) |
| 258 | parser.add_argument("--ddim_steps", type=int, default=50, help="steps of ddim if positive, otherwise use DDPM",) |
| 259 | parser.add_argument("--ddim_eta", type=float, default=1.0, help="eta for ddim sampling (0.0 yields deterministic sampling)",) |
| 260 | parser.add_argument("--bs", type=int, default=1, help="batch size for inference") |
| 261 | parser.add_argument("--height", type=int, default=512, help="image height, in pixel space") |
| 262 | parser.add_argument("--width", type=int, default=512, help="image width, in pixel space") |
| 263 | parser.add_argument("--unconditional_guidance_scale", type=float, default=1.0, help="prompt classifier-free guidance") |
| 264 | parser.add_argument("--seed", type=int, default=20230211, help="seed for seed_everything") |
| 265 | parser.add_argument("--cond_tau", type=float, default=1.0, help="",) |
| 266 | parser.add_argument("--target_size", type=int, default=None, help="", nargs="+") |
| 267 | parser.add_argument("--lora", type=bool, default=False, help="is use lora",) |
| 268 | parser.add_argument("--lora_scale", type=float, default=None, help="lora_scale",) |
| 269 | parser.add_argument("--pretrain", type=str, default='checkpoints/model_512/model-001.ckpt', help="pretrain model path") |
| 270 | return parser |
| 271 | |
| 272 | |
| 273 | if __name__ == '__main__': |