| 30 | |
| 31 | |
| 32 | def parse_args(): |
| 33 | argparser = argparse.ArgumentParser() |
| 34 | argparser.add_argument("--model_type", default='Llava', type=str, help='the number of results from search engine') |
| 35 | argparser.add_argument("--model_path", default='lmms-lab/llava-onevision-qwen2-7b-ov', type=str, help='the number of results from search engine') |
| 36 | argparser.add_argument("--world-size", type=int, default=1) |
| 37 | argparser.add_argument("--rank", type=int, default=0) |
| 38 | argparser.add_argument("--brief_result_num", default=8, type=int) |
| 39 | argparser.add_argument("--fullpage_num", default=1, type=int) |
| 40 | argparser.add_argument("--save_path", default='output/end2end/debug', type=str) |
| 41 | argparser.add_argument("--save_middle_results", type=str, default='') |
| 42 | argparser.add_argument("--generation_args_path", type=str, default='customs/generation_args.json', help='LMM generation parameters, should be a json') |
| 43 | argparser.add_argument("--verbose", action='store_true', default=False,) |
| 44 | return argparser.parse_args() |
| 45 | |
| 46 | args = parse_args() |
| 47 | |