| 22 | |
| 23 | |
| 24 | def parse_args(): |
| 25 | argparser = argparse.ArgumentParser() |
| 26 | argparser.add_argument("--model_type", default='Llava', type=str, help='the number of results from search engine') |
| 27 | 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') |
| 28 | argparser.add_argument("--world-size", type=int, default=1) |
| 29 | argparser.add_argument("--rank", type=int, default=0) |
| 30 | argparser.add_argument("--save_path", default='output/summarization/debug', type=str) |
| 31 | argparser.add_argument("--generation_args_path", type=str, default='customs/generation_args.json', help='LMM generation parameters, should be a json') |
| 32 | return argparser.parse_args() |
| 33 | |
| 34 | args = parse_args() |
| 35 | |