()
| 133 | |
| 134 | |
| 135 | def main(): |
| 136 | parser = argparse.ArgumentParser() |
| 137 | |
| 138 | parser.add_argument( |
| 139 | "--prompt", |
| 140 | type=str, |
| 141 | nargs="?", |
| 142 | default="a painting of a virus monster playing guitar", |
| 143 | help="the prompt to render" |
| 144 | ) |
| 145 | parser.add_argument( |
| 146 | "--outdir", |
| 147 | type=str, |
| 148 | nargs="?", |
| 149 | help="dir to write results to", |
| 150 | default="outputs/txt2img-samples" |
| 151 | ) |
| 152 | parser.add_argument( |
| 153 | "--skip_grid", |
| 154 | action='store_true', |
| 155 | help="do not save a grid, only individual samples. Helpful when evaluating lots of samples", |
| 156 | ) |
| 157 | parser.add_argument( |
| 158 | "--skip_save", |
| 159 | action='store_true', |
| 160 | help="do not save individual samples. For speed measurements.", |
| 161 | ) |
| 162 | parser.add_argument( |
| 163 | "--ddim_steps", |
| 164 | type=int, |
| 165 | default=200, |
| 166 | help="number of ddim sampling steps", |
| 167 | ) |
| 168 | parser.add_argument( |
| 169 | "--plms", |
| 170 | action='store_true', |
| 171 | help="use plms sampling", |
| 172 | ) |
| 173 | parser.add_argument( |
| 174 | "--laion400m", |
| 175 | action='store_true', |
| 176 | help="uses the LAION400M model", |
| 177 | ) |
| 178 | parser.add_argument( |
| 179 | "--fixed_code", |
| 180 | action='store_true', |
| 181 | help="if enabled, uses the same starting code across samples ", |
| 182 | ) |
| 183 | parser.add_argument( |
| 184 | "--ddim_eta", |
| 185 | type=float, |
| 186 | default=1.0, |
| 187 | help="ddim eta (eta=0.0 corresponds to deterministic sampling", |
| 188 | ) |
| 189 | parser.add_argument( |
| 190 | "--n_iter", |
| 191 | type=int, |
| 192 | default=1, |
no test coverage detected