chore: Add comments for each file extension type

This commit is contained in:
teleprint-me 2024-05-25 19:12:16 -04:00
parent da72554f58
commit fcd20ab9e9
No known key found for this signature in database
GPG key ID: B0D11345E65C4D48

View file

@ -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] = {