(parser)
| 52 | return module |
| 53 | |
| 54 | def get_args(parser): |
| 55 | parser.add_argument('--cfg', type=str, help='config file') |
| 56 | parser.add_argument('--base_model', type=str, default=None, help='model name in the hub or local path') |
| 57 | parser.add_argument('--llama_type', type=str, default=None, help='choose from llama2 or llama3') |
| 58 | parser.add_argument('--project_path', type=str, default="", help='project path') |
| 59 | |
| 60 | return parser.parse_args() |
| 61 | |
| 62 | if __name__ == "__main__": |
| 63 | parser = argparse.ArgumentParser() |