re-add type hint

Co-authored-by: compilade <git@compilade.net>
This commit is contained in:
Christian Zhou-Zheng 2024-06-06 08:27:25 -04:00 committed by GitHub
parent ce7e6985d2
commit 706bd69023
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -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,