()
| 3 | import torch |
| 4 | |
| 5 | def args(): |
| 6 | import argparse |
| 7 | parser = argparse.ArgumentParser() |
| 8 | parser.add_argument("--adapter_dir", type=str, required=True) |
| 9 | parser.add_argument("--full_model_dir", type=str, required=True) |
| 10 | parser.add_argument("--base_model_path_or_name", type=str, required=True) |
| 11 | return parser.parse_args() |
| 12 | |
| 13 | def main(): |
| 14 | script_args = args() |