swkimx97 commited on
Commit
596ab6d
·
verified ·
1 Parent(s): 3401a8b

Update sample_finetune.py

Browse files
Files changed (1) hide show
  1. 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 = "microsoft/Phi-3-mini-4k-instruct"
130
- # checkpoint_path = "microsoft/Phi-3-mini-128k-instruct"
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,