vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

convert.py : fix Enum to IntEnum

Browse source 
ggml-ci
This commit is contained in:
Georgi Gerganov 2023-08-22 16:28:30 +03:00
parent 32fc925ea8
commit 34151c9d2b
No known key found for this signature in database
GPG key ID: 449E073F9DC10735
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
convert.py
View file

@ -72,7 +72,7 @@ SAFETENSORS_DATA_TYPES: Dict[str, DataType] = {
# TODO: match this with `llama_ftype`
# TODO: rename to LLAMAFileType
# TODO: move to `gguf.py`
class GGMLFileType(enum.Enum):
class GGMLFileType(enum.IntEnum):
AllF32 = 0
MostlyF16 = 1 # except 1d tensors