Skip to content

Commit 454add4

Browse files
committed
Add model_type to gguf utility
1 parent 4250770 commit 454add4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

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'] | 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', 'mmproj'] | 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)