Update sample_finetune.py
#70
by
amcguinness
- opened
- 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"
|