From e833680ac6ef9f13c553d8c3101a45a3250446d2 Mon Sep 17 00:00:00 2001 From: brian khuu Date: Fri, 19 Jul 2024 00:12:55 +1000 Subject: [PATCH] convert-*.py: add general.name kv override --- gguf-py/gguf/metadata.py | 1 + 1 file changed, 1 insertion(+) diff --git a/gguf-py/gguf/metadata.py b/gguf-py/gguf/metadata.py index be297f2426f78..bac6ebfb3777a 100644 --- a/gguf-py/gguf/metadata.py +++ b/gguf-py/gguf/metadata.py @@ -62,6 +62,7 @@ def load(metadata_override_path: Optional[Path] = None, model_path: Optional[Pat # This is based on LLM_KV_NAMES mapping in llama.cpp metadata_override = Metadata.load_metadata_override(metadata_override_path) + metadata.name = metadata_override.get(Keys.General.NAME, metadata.name) metadata.author = metadata_override.get(Keys.General.AUTHOR, metadata.author) metadata.version = metadata_override.get(Keys.General.VERSION, metadata.version) metadata.organization = metadata_override.get(Keys.General.ORGANIZATION, metadata.organization)