Procházet zdrojové kódy

gguf-py : do not use internal numpy types (#7472)

compilade před 1 rokem
rodič
revize
7d0e23d72e
1 změnil soubory, kde provedl 1 přidání a 2 odebrání
  1. 1 2
      gguf-py/gguf/lazy.py

+ 1 - 2
gguf-py/gguf/lazy.py

@@ -6,7 +6,6 @@ from typing import Any, Callable
 from collections import deque
 from collections import deque
 
 
 import numpy as np
 import numpy as np
-from numpy._typing import _Shape
 from numpy.typing import DTypeLike
 from numpy.typing import DTypeLike
 
 
 
 
@@ -219,7 +218,7 @@ class LazyNumpyTensor(LazyBase):
     _tensor_type = np.ndarray
     _tensor_type = np.ndarray
 
 
     @classmethod
     @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,
         # The initial idea was to use np.nan as the fill value,
         # but non-float types like np.int16 can't use that.
         # but non-float types like np.int16 can't use that.
         # So zero it is.
         # So zero it is.