Skip to content

Commit 8e0fc8c

Browse files
committed
Convert script uses correct ckpt
1 parent 147805a commit 8e0fc8c

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

examples/models/phi-4-mini/convert_weights.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -67,12 +67,11 @@ def main():
6767
checkpointer = FullModelHFCheckpointer(
6868
checkpoint_dir=args.input_dir,
6969
checkpoint_files=[
70-
"model-00001-of-00003.safetensors",
71-
"model-00002-of-00003.safetensors",
72-
"model-00003-of-00003.safetensors",
70+
"model-00001-of-00002.safetensors",
71+
"model-00002-of-00002.safetensors",
7372
],
7473
output_dir=".",
75-
model_type="PHI4_MINI",
74+
model_type="PHI3_MINI",
7675
)
7776

7877
print("Loading checkpoint...")

0 commit comments

Comments
 (0)