From dc5cf5fd825a0ffa4964fdebc4f9b42444d39ca2 Mon Sep 17 00:00:00 2001 From: Christian Zhou-Zheng <59622928+christianazinn@users.noreply.github.com> Date: Fri, 7 Jun 2024 17:26:30 -0400 Subject: [PATCH] Update gguf-py/gguf/gguf_writer_split.py Co-authored-by: compilade --- gguf-py/gguf/gguf_writer_split.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/gguf-py/gguf/gguf_writer_split.py b/gguf-py/gguf/gguf_writer_split.py index c11567999..a419512ce 100644 --- a/gguf-py/gguf/gguf_writer_split.py +++ b/gguf-py/gguf/gguf_writer_split.py @@ -25,8 +25,8 @@ from .constants import Keys SHARD_NAME_FORMAT = "{:s}-{:05d}-of-{:05d}.gguf" METADATA_ONLY_INDICATOR = -1 -KVTempData: TypeAlias = dict[str, tuple[Any, GGUFValueType]] # {key: (value, type)} -TensorTempData: TypeAlias = tuple[str, np.ndarray[Any, Any], GGMLQuantizationType] # (tensor name, tensor data, tensor dtype) +KVTempData: TypeAlias = dict[str, tuple[Any, GGUFValueType | None]] # {key: (value, type)} +TensorTempData: TypeAlias = tuple[str, np.ndarray[Any, Any], GGMLQuantizationType | None] # (tensor name, tensor data, tensor dtype) @dataclass