From fcd20ab9e9fbca5e50f6bf60033d62c61dde13c8 Mon Sep 17 00:00:00 2001
From: teleprint-me <77757836+teleprint-me@users.noreply.github.com>
Date: Sat, 25 May 2024 19:12:16 -0400
Subject: [PATCH] chore: Add comments for each file extension type

---
 gguf-py/gguf/constants.py | 16 ++++++++--------
 1 file changed, 8 insertions(+), 8 deletions(-)

diff --git a/gguf-py/gguf/constants.py b/gguf-py/gguf/constants.py
index 382f0fda2..4430749d7 100644
--- a/gguf-py/gguf/constants.py
+++ b/gguf-py/gguf/constants.py
@@ -969,14 +969,14 @@ VOCAB_TYPE_NAMES: dict[VocabType, str] = {
 # Model File Types
 #
 class ModelFileType(IntEnum):
-    UNK         = auto()
-    PT          = auto()
-    PTH         = auto()
-    BIN         = auto()
-    SAFETENSORS = auto()
-    JSON        = auto()
-    MODEL       = auto()
-    GGUF        = auto()
+    NON         = auto()  # undefined
+    PT          = auto()  # torch
+    PTH         = auto()  # torch
+    BIN         = auto()  # torch
+    SAFETENSORS = auto()  # safetensors
+    JSON        = auto()  # transformers/tokenizers
+    MODEL       = auto()  # sentencepiece
+    GGUF        = auto()  # ggml/llama.cpp
 
 
 MODEL_FILE_TYPE_NAMES: dict[ModelFileType, str] = {