Commit
·
75955e4
1
Parent(s):
58cd8b9
flavor
Browse files
sft.py
CHANGED
|
@@ -6,9 +6,10 @@ parser = argparse.ArgumentParser(description="Demo script for the model.")
|
|
| 6 |
|
| 7 |
parser.add_argument("--model", type=str)
|
| 8 |
parser.add_argument("--dataset", type=str)
|
|
|
|
| 9 |
|
| 10 |
args = parser.parse_args()
|
| 11 |
-
print(f"Model: {args.model}, Dataset: {args.dataset}")
|
| 12 |
|
| 13 |
# Run another Python script
|
| 14 |
subprocess.run(["trl", "sft", "--config", "configs/Qwen3-4B-Base_a10g-small.yaml"])
|
|
|
|
| 6 |
|
| 7 |
parser.add_argument("--model", type=str)
|
| 8 |
parser.add_argument("--dataset", type=str)
|
| 9 |
+
parser.add_argument("--flavor", type=str)
|
| 10 |
|
| 11 |
args = parser.parse_args()
|
| 12 |
+
print(f"Model: {args.model}, Dataset: {args.dataset}, Flavor: {args.flavor}")
|
| 13 |
|
| 14 |
# Run another Python script
|
| 15 |
subprocess.run(["trl", "sft", "--config", "configs/Qwen3-4B-Base_a10g-small.yaml"])
|