swkimx97
/

Phi-3-mini-4k-instruct

Model card Files Files and versions

swkimx97 commited on Dec 1, 2024

Commit

596ab6d

·

verified ·

1 Parent(s): 3401a8b

Update sample_finetune.py

Files changed (1) hide show

sample_finetune.py +2 -2

sample_finetune.py CHANGED Viewed

@@ -126,8 +126,8 @@ logger.info(f"PEFT parameters {peft_conf}")
 ################
 # Model Loading
 ################
-checkpoint_path = "microsoft/Phi-3-mini-4k-instruct"
-# checkpoint_path = "microsoft/Phi-3-mini-128k-instruct"
 model_kwargs = dict(
     use_cache=False,
     trust_remote_code=True,

 ################
 # Model Loading
 ################
+checkpoint_path = "swkimx97/Phi-3-mini-4k-instruct"
+# checkpoint_path = "swkimx97/Phi-3-mini-128k-instruct"
 model_kwargs = dict(
     use_cache=False,
     trust_remote_code=True,