We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 15a3332 commit 77178eeCopy full SHA for 77178ee
gguf-py/gguf/gguf_writer.py
@@ -204,7 +204,7 @@ def add_tensor_info(
204
for i in range(n_dims):
205
self.ti_data += self._pack("Q", tensor_shape[n_dims - 1 - i])
206
if raw_dtype is None:
207
- if tensor_shape == np.float32:
+ if tensor_dtype == np.float32:
208
dtype = GGMLQuantizationType.F32
209
elif tensor_dtype == np.float16:
210
dtype = GGMLQuantizationType.F16
0 commit comments