Update sample_finetune.py
Browse files- sample_finetune.py +2 -2
sample_finetune.py
CHANGED
@@ -126,8 +126,8 @@ logger.info(f"PEFT parameters {peft_conf}")
|
|
126 |
################
|
127 |
# Model Loading
|
128 |
################
|
129 |
-
checkpoint_path = "
|
130 |
-
# checkpoint_path = "
|
131 |
model_kwargs = dict(
|
132 |
use_cache=False,
|
133 |
trust_remote_code=True,
|
|
|
126 |
################
|
127 |
# Model Loading
|
128 |
################
|
129 |
+
checkpoint_path = "swkimx97/Phi-3-mini-4k-instruct"
|
130 |
+
# checkpoint_path = "swkimx97/Phi-3-mini-128k-instruct"
|
131 |
model_kwargs = dict(
|
132 |
use_cache=False,
|
133 |
trust_remote_code=True,
|