diff --git a/gguf.py b/gguf.py index 1f24cb9d5..f1e2c69a2 100644 --- a/gguf.py +++ b/gguf.py @@ -62,7 +62,7 @@ class GGUFWriter: def __init__(self, fout: IO): self.fout = fout self.offset_tensor = 0 - self.tensors = [] + self.tensors: List[np.ndarray] = [] def write_header(self, tensor_count: int, metadata_kv_count: int): self.fout.write(struct.pack("