CS348Project/llama.cpp
5
0
Fork 0
mirror of https://github.com/ggml-org/llama.cpp.git synced 2025-10-29 08:41:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
compilade
2024-07-09 01:04:49 -04:00
committed by GitHub
parent 7fdb6f73e3
commit 7d0e23d72e
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
gguf-py/gguf/lazy.py
View File

@@ -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.