train / sft.py
qgallouedec's picture
qgallouedec HF Staff
sft directly exec
e43de7f
raw
history blame
284 Bytes
#!/usr/bin/env python3
import argparse
parser = argparse.ArgumentParser(description="Demo script for the model.")
parser.add_argument("--model", type=str)
parser.add_argument("--dataset", type=str)
args = parser.parse_args()
print(f"Model: {args.model}, Dataset: {args.dataset}")