Skip to content

Commit 9764773

Browse files
committed
convert-*.py: model name should use folder name over hyper parameter metadata
1 parent 9626f4f commit 9764773

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

convert-hf-to-gguf.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -138,12 +138,15 @@ def get_model_name(metadata, hyperparameters, dir_model, model_arch):
138138
if metadata is not None and metadata.name is not None:
139139
# Explicit Metadata Was Provided By User
140140
return metadata.name
141-
elif hyperparameters is not None and "model_type" in hyperparameters:
142-
# Hugging Face Hyperparameter Model Type Already Provided
143-
return hyperparameters["model_type"]
144141
elif dir_model is not None and dir_model.name is not None:
145142
# Use directory folder name
146143
return dir_model.name
144+
elif hyperparameters is not None and "_name_or_path" in hyperparameters:
145+
# Hugging Face Hyperparameter Model Name or Model Folder Name is Provided
146+
return hyperparameters["_name_or_path"]
147+
elif hyperparameters is not None and "model_type" in hyperparameters:
148+
# Hugging Face Hyperparameter Model Type is Provided
149+
return hyperparameters["model_type"]
147150
else:
148151
return gguf.MODEL_ARCH_NAMES[model_arch]
149152
self.model_name = get_model_name(self.metadata, self.hparams, self.dir_model, self.model_arch)

0 commit comments

Comments
 (0)