()
| 41 | |
| 42 | |
| 43 | def parse_args(): |
| 44 | parser = argparse.ArgumentParser('', add_help=False) |
| 45 | parser.add_argument('--path', help='path of folder to checkpoints', |
| 46 | type=str) |
| 47 | parser.add_argument('--newtoken', help='number of new tokens in the checkpoint', default=1, |
| 48 | type=int) |
| 49 | |
| 50 | return parser.parse_args() |
| 51 | |
| 52 | |
| 53 | if __name__ == "__main__": |