Commit
·
ebf76b5
1
Parent(s):
ad5dd74
basic model and dataset parsing
Browse files
sft.py
CHANGED
|
@@ -1 +1,10 @@
|
|
| 1 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import argparse
|
| 2 |
+
|
| 3 |
+
parser = argparse.ArgumentParser(description="SFT training script")
|
| 4 |
+
|
| 5 |
+
parser.add_argument("--model", type=str)
|
| 6 |
+
parser.add_argument("--dataset", type=str)
|
| 7 |
+
|
| 8 |
+
args = parser.parse_args()
|
| 9 |
+
|
| 10 |
+
print(f"Model: {args.model}, Dataset: {args.dataset}")
|