From 8334b5becb80753256c825d2a37ba94a5fc97d0a Mon Sep 17 00:00:00 2001 From: Francis Couture-Harpin Date: Wed, 22 May 2024 14:29:50 -0400 Subject: [PATCH] gguf-py : do not use internal numpy types --- gguf-py/gguf/lazy.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/gguf-py/gguf/lazy.py b/gguf-py/gguf/lazy.py index 1167335b8..bad5cf8a9 100644 --- a/gguf-py/gguf/lazy.py +++ b/gguf-py/gguf/lazy.py @@ -6,7 +6,6 @@ from typing import Any, Callable from collections import deque import numpy as np -from numpy._typing import _Shape from numpy.typing import DTypeLike @@ -216,7 +215,7 @@ class LazyNumpyTensor(LazyBase): _tensor_type = np.ndarray @classmethod - def meta_with_dtype_and_shape(cls, dtype: DTypeLike, shape: _Shape) -> np.ndarray[Any, Any]: + def meta_with_dtype_and_shape(cls, dtype: DTypeLike, shape: tuple[int, ...]) -> np.ndarray[Any, Any]: # The initial idea was to use np.nan as the fill value, # but non-float types like np.int16 can't use that. # So zero it is.