]> git.djapps.eu Git - pkg/ggml/sources/llama.cpp/commitdiff
gguf-py : do not use internal numpy types (#7472)
authorcompilade <redacted>
Tue, 9 Jul 2024 05:04:49 +0000 (01:04 -0400)
committerGitHub <redacted>
Tue, 9 Jul 2024 05:04:49 +0000 (01:04 -0400)
gguf-py/gguf/lazy.py

index c50124cd912d40fc1936e1d39347d8d171134dd4..6e266f34f76ace877eca3be5ff096c7e4c72dcb4 100644 (file)
@@ -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
 
 
@@ -219,7 +218,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.