From 706bd69023a5d93f678a0d59337a24d2565b4d68 Mon Sep 17 00:00:00 2001 From: Christian Zhou-Zheng <59622928+christianazinn@users.noreply.github.com> Date: Thu, 6 Jun 2024 08:27:25 -0400 Subject: [PATCH] re-add type hint Co-authored-by: compilade --- convert-hf-to-gguf.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/convert-hf-to-gguf.py b/convert-hf-to-gguf.py index b6fd4bc4b..643602883 100644 --- a/convert-hf-to-gguf.py +++ b/convert-hf-to-gguf.py @@ -2857,7 +2857,7 @@ def main() -> None: split_arguments = gguf.SplitArguments(args) - ftype_map = { + ftype_map: dict[str, gguf.LlamaFileType] = { "f32": gguf.LlamaFileType.ALL_F32, "f16": gguf.LlamaFileType.MOSTLY_F16, "bf16": gguf.LlamaFileType.MOSTLY_BF16,