| 19 | import datasets |
| 20 | |
| 21 | def parse_args(): |
| 22 | argparser = argparse.ArgumentParser() |
| 23 | argparser.add_argument("--model_type", default='Llava', type=str, help='the number of results from search engine') |
| 24 | argparser.add_argument("--model_path", default='/data1/zrr/jdz/models/llava-next-interleave-qwen-7b', type=str, help='the number of results from search engine') |
| 25 | argparser.add_argument("--world-size", type=int, default=1) |
| 26 | argparser.add_argument("--rank", type=int, default=0) |
| 27 | argparser.add_argument("--brief_result_num", default=8, type=int) |
| 28 | argparser.add_argument("--fullpage_num", default=1, type=int) |
| 29 | argparser.add_argument("--save_path", default='output/rerank/debug', type=str) |
| 30 | argparser.add_argument("--generation_args_path", type=str, default='customs/generation_args.json', help='LMM generation parameters, should be a json') |
| 31 | return argparser.parse_args() |
| 32 | |
| 33 | |
| 34 | args = parse_args() |