generation api arguments
(parser)
| 273 | return parser |
| 274 | |
| 275 | def add_generation_api_args(parser): |
| 276 | """generation api arguments""" |
| 277 | |
| 278 | group = parser.add_argument_group('api', 'api configurations') |
| 279 | |
| 280 | group.add_argument('--img_folder_path', default='image/') |
| 281 | group.add_argument('--input_folder_path', default='input/') |
| 282 | group.add_argument('--input_rec_path', default='input/') |
| 283 | group.add_argument('--check_mode', default='code') |
| 284 | group.add_argument('--time_interval', default=10) |
| 285 | group.add_argument('--device', default=None) |
| 286 | |
| 287 | return parser |
| 288 | |
| 289 | def add_sparse_args(parser): |
| 290 | """sparse attention arguments.""" |