From 34151c9d2baf7accdbc6f75be4939333e5bfd61a Mon Sep 17 00:00:00 2001 From: Georgi Gerganov Date: Tue, 22 Aug 2023 16:28:30 +0300 Subject: [PATCH] convert.py : fix Enum to IntEnum ggml-ci --- convert.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/convert.py b/convert.py index 635c3e275..71978d671 100644 --- a/convert.py +++ b/convert.py @@ -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