| 28 | |
| 29 | |
| 30 | def parse_arguments(): |
| 31 | parser = argparse.ArgumentParser() |
| 32 | parser.add_argument('--model_name', default=None, |
| 33 | choices=[ |
| 34 | 'gpt4', |
| 35 | 'gpt3.5', |
| 36 | 'geminipro', |
| 37 | 'palm2', |
| 38 | 'qwenchat', |
| 39 | 'yichat', |
| 40 | 'chatglm3', |
| 41 | 'baichuan2chat', |
| 42 | 'internlm2chat', |
| 43 | 'discmedllm', |
| 44 | 'bianque2', |
| 45 | 'pulse', |
| 46 | 'huatuogpt2', |
| 47 | 'taiyillm', |
| 48 | 'wingpt2', |
| 49 | 'spark3', |
| 50 | 'bluelmchat', |
| 51 | 'claude3' |
| 52 | ], type=str) |
| 53 | parser.add_argument('--model_path', default=None, type=str) |
| 54 | parser.add_argument('--cache_dir', default=None, type=str) |
| 55 | parser.add_argument('--api_key', default=None, type=str) |
| 56 | parser.add_argument('--api_secret', default=None, type=str) # only for spark3 |
| 57 | parser.add_argument('--app_id', default=None, type=str) # only for spark3 |
| 58 | parser.add_argument('--data_load_name', default=None, type=str) |
| 59 | parser.add_argument('--result_save_name', default=None, type=str) |
| 60 | parser.add_argument('--log_save_name', default=None, type=str) |
| 61 | args = parser.parse_args() |
| 62 | |
| 63 | return args |
| 64 | |
| 65 | |
| 66 | def generate_hospital_guide(example): |