| import argparse | |
| import trl | |
| parser = argparse.ArgumentParser(description="SFT training script") | |
| parser.add_argument("--model", type=str) | |
| parser.add_argument("--dataset", type=str) | |
| args = parser.parse_args() | |
| print(f"Model: {args.model}, Dataset: {args.dataset}") | |