Sampling configurations
(parser)
| 29 | |
| 30 | |
| 31 | def add_sampling_config_args(parser): |
| 32 | """Sampling configurations""" |
| 33 | |
| 34 | group = parser.add_argument_group("sampling", "Sampling Configurations") |
| 35 | group.add_argument("--output-dir", type=str, default="samples") |
| 36 | group.add_argument("--input-dir", type=str, default=None) |
| 37 | group.add_argument("--input-type", type=str, default="cli") |
| 38 | group.add_argument("--input-file", type=str, default="input.txt") |
| 39 | group.add_argument("--final-size", type=int, default=2048) |
| 40 | group.add_argument("--sdedit", action="store_true") |
| 41 | group.add_argument("--grid-num-rows", type=int, default=1) |
| 42 | group.add_argument("--force-inference", action="store_true") |
| 43 | group.add_argument("--lcm_steps", type=int, default=None) |
| 44 | group.add_argument("--sampling-num-frames", type=int, default=32) |
| 45 | group.add_argument("--sampling-fps", type=int, default=8) |
| 46 | group.add_argument("--only-save-latents", type=bool, default=False) |
| 47 | group.add_argument("--only-log-video-latents", type=bool, default=False) |
| 48 | group.add_argument("--latent-channels", type=int, default=32) |
| 49 | group.add_argument("--image2video", action="store_true") |
| 50 | |
| 51 | return parser |
| 52 | |
| 53 | |
| 54 |