Text generate arguments.
(parser)
| 203 | |
| 204 | |
| 205 | def add_text_generate_args(parser): |
| 206 | """Text generate arguments.""" |
| 207 | |
| 208 | group = parser.add_argument_group('Text generation', 'configurations') |
| 209 | group.add_argument("--temperature", type=float, default=1.0) |
| 210 | group.add_argument("--top_p", type=float, default=0.0) |
| 211 | group.add_argument("--top_k", type=int, default=0) |
| 212 | # group.add_argument("--out-seq-length", type=int, default=256) |
| 213 | group.add_argument("--generation-task", type=str, |
| 214 | default='text2image', |
| 215 | choices=['text2image', |
| 216 | 'image2text', |
| 217 | 'super-resolution', |
| 218 | 'low-level super-resolution', |
| 219 | 'post-selection', |
| 220 | 'raw' |
| 221 | ], |
| 222 | help='what type of inference task to use') |
| 223 | group.add_argument('--input-source', type=str, default='interactive', |
| 224 | help='what input mode to use, interactive or path') |
| 225 | group.add_argument('--output-path', type=str, default='./samples', |
| 226 | help='path to place the generated samples') |
| 227 | group.add_argument('--debug', action='store_true', |
| 228 | help='Debug will merge all outputs.') |
| 229 | group.add_argument('--with-id', action='store_true', |
| 230 | help='If each line is prepended with an id.') |
| 231 | group.add_argument('--max-inference-batch-size', type=int, default=12) |
| 232 | return parser |
| 233 | |
| 234 | |
| 235 | def add_data_args(parser): |