Skip to content

Commit c7a926f

Browse files
cebtenzzremofosyne
authored andcommitted
convert : get general.name from model dir, not its parent (ggml-org#5615)
Co-authored-by: Brian <mofosyne@gmail.com>
1 parent d0a9c31 commit c7a926f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

convert.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1109,7 +1109,7 @@ def add_meta_model(self, params: Params, metadata: Metadata) -> None:
11091109
if metadata is not None and metadata.name is not None:
11101110
name = metadata.name
11111111
elif params.path_model is not None:
1112-
name = str(params.path_model.parent).split("/")[-1]
1112+
name = params.path_model.name
11131113
elif params.n_ctx == 4096:
11141114
# Heuristic detection of LLaMA v2 model
11151115
name = "LLaMA v2"

0 commit comments

Comments
 (0)