Skip to content

Commit 02fe159

Browse files
committed
Use mmproj- prefix instead of suffix
1 parent 454add4 commit 02fe159

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

convert_hf_to_gguf.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1503,8 +1503,8 @@ def prepare_metadata(self, vocab_only: bool):
15031503
output_type: str = self.ftype.name.partition("_")[2]
15041504

15051505
if self.fname_out.is_dir():
1506-
fname_default: str = gguf.naming_convention(self.metadata.name, self.metadata.basename, self.metadata.finetune, self.metadata.version, size_label=None, output_type=output_type, model_type="mmproj")
1507-
self.fname_out = self.fname_out / f"{fname_default}.gguf"
1506+
fname_default: str = gguf.naming_convention(self.metadata.name, self.metadata.basename, self.metadata.finetune, self.metadata.version, size_label=None, output_type=output_type, model_type=None)
1507+
self.fname_out = self.fname_out / f"mmproj-{fname_default}.gguf"
15081508
else:
15091509
self.fname_out = self.fname_out.parent / gguf.fill_templated_filename(self.fname_out.name, output_type)
15101510

gguf-py/gguf/utility.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,7 @@ def size_label(total_params: int, shared_params: int, expert_params: int, expert
5050
return size_class
5151

5252

53-
def naming_convention(model_name: str | None, base_name: str | None, finetune_string: str | None, version_string: str | None, size_label: str | None, output_type: str | None, model_type: Literal['vocab', 'LoRA', 'mmproj'] | None = None) -> str:
53+
def naming_convention(model_name: str | None, base_name: str | None, finetune_string: str | None, version_string: str | None, size_label: str | None, output_type: str | None, model_type: Literal['vocab', 'LoRA'] | None = None) -> str:
5454
# Reference: https://github.com/ggml-org/ggml/blob/master/docs/gguf.md#gguf-naming-convention
5555

5656
if base_name is not None:

0 commit comments

Comments
 (0)