Skip to content

Commit fa95ec3

Browse files
committed
convert-*.py: hyperparameters takes precidence over folder name
1 parent 9764773 commit fa95ec3

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

convert-hf-to-gguf.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -138,15 +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 dir_model is not None and dir_model.name is not None:
142-
# Use directory folder name
143-
return dir_model.name
144141
elif hyperparameters is not None and "_name_or_path" in hyperparameters:
145142
# Hugging Face Hyperparameter Model Name or Model Folder Name is Provided
146143
return hyperparameters["_name_or_path"]
147144
elif hyperparameters is not None and "model_type" in hyperparameters:
148145
# Hugging Face Hyperparameter Model Type is Provided
149146
return hyperparameters["model_type"]
147+
elif dir_model is not None and dir_model.name is not None:
148+
# Use directory folder name
149+
return dir_model.name
150150
else:
151151
return gguf.MODEL_ARCH_NAMES[model_arch]
152152
self.model_name = get_model_name(self.metadata, self.hparams, self.dir_model, self.model_arch)

0 commit comments

Comments
 (0)