File size: 414 Bytes
e43de7f 411ce45 c5d1a86 411ce45 c5d1a86 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
#!/usr/bin/env python3
import argparse
import subprocess
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}")
# Run another Python script
subprocess.run(["trl", "sft", "--config", "configs/Qwen3-4B-Base_a10g-small.yaml"])
|