Update sample_finetune.py (#70)
Browse files- Update sample_finetune.py (bd1e66a6743baac0bfd87a6c51c5a56a97776630)
Co-authored-by: Adam McGuinness <[email protected]>
- 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"
|