amcguinness
commited on
Commit
•
bd1e66a
1
Parent(s):
bbd531d
Update sample_finetune.py
Browse files- sample_finetune.py +1 -1
sample_finetune.py
CHANGED
@@ -124,7 +124,7 @@ logger.info(f"PEFT parameters {peft_conf}")
|
|
124 |
|
125 |
|
126 |
################
|
127 |
-
#
|
128 |
################
|
129 |
checkpoint_path = "microsoft/Phi-3-mini-4k-instruct"
|
130 |
# checkpoint_path = "microsoft/Phi-3-mini-128k-instruct"
|
|
|
124 |
|
125 |
|
126 |
################
|
127 |
+
# Model Loading
|
128 |
################
|
129 |
checkpoint_path = "microsoft/Phi-3-mini-4k-instruct"
|
130 |
# checkpoint_path = "microsoft/Phi-3-mini-128k-instruct"
|