| #!/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) | |
| parser.add_argument("--flavor", type=str) | |
| args = parser.parse_args() | |
| print(f"Model: {args.model}, Dataset: {args.dataset}, Flavor: {args.flavor}") | |
| # Run another Python script | |
| subprocess.run(["trl", "sft", "--config", "configs/Qwen3-4B-Base_a10g-small.yaml"]) | |