Skip to content

Commit 147805a

Browse files
committed
Lint
1 parent a12912a commit 147805a

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

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

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -66,7 +66,11 @@ def main():
6666

6767
checkpointer = FullModelHFCheckpointer(
6868
checkpoint_dir=args.input_dir,
69-
checkpoint_files=["model-00001-of-00003.safetensors", "model-00002-of-00003.safetensors", "model-00003-of-00003.safetensors"],
69+
checkpoint_files=[
70+
"model-00001-of-00003.safetensors",
71+
"model-00002-of-00003.safetensors",
72+
"model-00003-of-00003.safetensors",
73+
],
7074
output_dir=".",
7175
model_type="PHI4_MINI",
7276
)

0 commit comments

Comments
 (0)