From 32e80e094cd7fbb5441c216839448c41e726fb4a Mon Sep 17 00:00:00 2001 From: brian khuu Date: Tue, 4 Jun 2024 00:28:16 +1000 Subject: [PATCH] convert-*.py: base_model is actually in spec for model cards --- gguf-py/gguf/metadata.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/gguf-py/gguf/metadata.py b/gguf-py/gguf/metadata.py index 3107e59cb..61b6a510f 100644 --- a/gguf-py/gguf/metadata.py +++ b/gguf-py/gguf/metadata.py @@ -51,8 +51,6 @@ class Metadata: metadata.name = model_card.get("model_name") if "base_model" in model_card: - # Not part of huggingface model card standard but notice some model creator using it - # such as TheBloke who would encode 'mistralai/Mixtral-8x7B-Instruct-v0.1' into base_model model_id = model_card.get("base_model") model_name_normal, organization_name, base_name, fine_tune, version_string, parameter_weight_class = Metadata.get_model_name_components(model_id)