From fa91d869cf0a0fa10f95cd1656554f9e054242d1 Mon Sep 17 00:00:00 2001 From: Brian Date: Fri, 19 Jul 2024 17:51:51 +1000 Subject: [PATCH] convert-*.py: add general.name kv override (#8571) --- 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)