Skip to content

Commit 15d7834

Browse files
committed
1 parent 6b455c6 commit 15d7834

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

build/builder.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -206,8 +206,11 @@ def _load_model_not_gguf(builder_args):
206206
else:
207207
checkpoint[key] = cps[0][key]
208208
else:
209+
path = builder_args.checkpoint_path
210+
if isinstance(builder_args.checkpoint_path, Path):
211+
path = str(path)
209212
checkpoint = torch.load(
210-
builder_args.checkpoint_path,
213+
path,
211214
map_location=builder_args.device,
212215
mmap=True,
213216
weights_only=True,

0 commit comments

Comments
 (0)