()
| 6 | import argparse |
| 7 | |
| 8 | def get_args(): |
| 9 | parser = argparse.ArgumentParser() |
| 10 | parser.add_argument("--base_model_name_or_path", type=str, default="bigcode/large-model") |
| 11 | parser.add_argument("--peft_model_path", type=str, default="/") |
| 12 | parser.add_argument("--push_to_hub", action="store_true", default=True) |
| 13 | |
| 14 | return parser.parse_args() |
| 15 | |
| 16 | def main(): |
| 17 | args = get_args() |