File size: 254 Bytes
ebf76b5
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
import argparse

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}")