Pinkstack commited on
Commit
1050914
·
verified ·
1 Parent(s): f817e22

Update sample_finetune.py

Browse files
Files changed (1) hide show
  1. sample_finetune.py +2 -2
sample_finetune.py CHANGED
@@ -128,10 +128,10 @@ logger.info(f"PEFT parameters {peft_conf}")
128
  ################
129
  # Model Loading
130
  ################
131
- checkpoint_path = "microsoft/Phi-4-mini-instruct"
132
  model_kwargs = dict(
133
  use_cache=False,
134
- trust_remote_code=True,
135
  attn_implementation="flash_attention_2", # loading the model with flash-attention support
136
  torch_dtype=torch.bfloat16,
137
  device_map=None
 
128
  ################
129
  # Model Loading
130
  ################
131
+ checkpoint_path = "Pinkstack/Phi-4-mini-6b-merge"
132
  model_kwargs = dict(
133
  use_cache=False,
134
+ trust_remote_code=False,
135
  attn_implementation="flash_attention_2", # loading the model with flash-attention support
136
  torch_dtype=torch.bfloat16,
137
  device_map=None