(args=None)
| 19 | save_dir = 'multiprocess_data' |
| 20 | |
| 21 | def parse_args(args=None): |
| 22 | parser = argparse.ArgumentParser() |
| 23 | parser.add_argument('--model', default="chatglm", type=str) |
| 24 | return parser.parse_args(args) |
| 25 | |
| 26 | def process_file(lines, rank, args): |
| 27 | def build_input(conversations, tokenizer, args): |