re-add type hint

Co-authored-by: compilade <[email protected]>
christianazinn · Jun 6, 2024 · 706bd69 · 706bd69
1 parent ce7e698
commit 706bd69
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,