File size: 265 Bytes
ebf76b5 e387537 ebf76b5 |
1 2 3 4 5 6 7 8 9 10 11 12 |
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}")
|