Update train.py
Browse files
train.py
CHANGED
|
@@ -11,7 +11,7 @@ phi = (1 + np.sqrt(5)) / 2
|
|
| 11 |
# -----------------------------
|
| 12 |
# Load model + dataset
|
| 13 |
# -----------------------------
|
| 14 |
-
model_name = "antonypamo/
|
| 15 |
dataset = load_dataset("antonypamo/savantorganized", data_files="data/corpus_balanced_phi93.jsonl")
|
| 16 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
| 17 |
model = AutoModelForMaskedLM.from_pretrained(model_name)
|
|
|
|
| 11 |
# -----------------------------
|
| 12 |
# Load model + dataset
|
| 13 |
# -----------------------------
|
| 14 |
+
model_name = "antonypamo/ProSavantEngine_Phi9_4"
|
| 15 |
dataset = load_dataset("antonypamo/savantorganized", data_files="data/corpus_balanced_phi93.jsonl")
|
| 16 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
| 17 |
model = AutoModelForMaskedLM.from_pretrained(model_name)
|