| 25 | |
| 26 | |
| 27 | def parse_args(): |
| 28 | argparser = argparse.ArgumentParser() |
| 29 | argparser.add_argument("--model_type", default='Llava', type=str, help='the number of results from search engine') |
| 30 | argparser.add_argument("--model_path", default='lmms-lab/llava-onevision-qwen2-7b-ov', type=str, help='the number of results from search engine') |
| 31 | argparser.add_argument("--world-size", type=int, default=1) |
| 32 | argparser.add_argument("--rank", type=int, default=0) |
| 33 | argparser.add_argument("--save_path", default='output/requery/debug', type=str) |
| 34 | argparser.add_argument("--generation_args_path", type=str, default='customs/generation_args.json', help='LMM generation parameters, should be a json') |
| 35 | return argparser.parse_args() |
| 36 | |
| 37 | args = parse_args() |
| 38 | |