()
| 296 | |
| 297 | |
| 298 | def main(): |
| 299 | parser = argparse.ArgumentParser() |
| 300 | |
| 301 | parser.add_argument( |
| 302 | "--prompt", |
| 303 | type=str, |
| 304 | nargs="?", |
| 305 | default="a painting of a virus monster playing guitar", |
| 306 | help="the prompt to render" |
| 307 | ) |
| 308 | parser.add_argument( |
| 309 | "--outdir", |
| 310 | type=str, |
| 311 | nargs="?", |
| 312 | help="dir to write results to", |
| 313 | default="outputs/txt2img-samples" |
| 314 | ) |
| 315 | parser.add_argument( |
| 316 | "--skip_grid", |
| 317 | action='store_true', |
| 318 | help="do not save a grid, only individual samples. Helpful when evaluating lots of samples", |
| 319 | ) |
| 320 | parser.add_argument( |
| 321 | "--skip_save", |
| 322 | action='store_true', |
| 323 | help="do not save individual samples. For speed measurements.", |
| 324 | ) |
| 325 | parser.add_argument( |
| 326 | "--ddim_steps", |
| 327 | type=int, |
| 328 | default=200, |
| 329 | help="number of ddim sampling steps", |
| 330 | ) |
| 331 | parser.add_argument( |
| 332 | "--plms", |
| 333 | action='store_true', |
| 334 | help="use plms sampling", |
| 335 | ) |
| 336 | parser.add_argument( |
| 337 | "--laion400m", |
| 338 | action='store_true', |
| 339 | help="uses the LAION400M model", |
| 340 | ) |
| 341 | parser.add_argument( |
| 342 | "--fixed_code", |
| 343 | action='store_true', |
| 344 | help="if enabled, uses the same starting code across samples ", |
| 345 | ) |
| 346 | parser.add_argument( |
| 347 | "--ddim_eta", |
| 348 | type=float, |
| 349 | default=1.0, |
| 350 | help="ddim eta (eta=0.0 corresponds to deterministic sampling", |
| 351 | ) |
| 352 | parser.add_argument( |
| 353 | "--n_iter", |
| 354 | type=int, |
| 355 | default=1, |
no test coverage detected