Update gguf-py/gguf/gguf_writer.py

Co-authored-by: compilade <[email protected]>
christianazinn · Jun 10, 2024 · e9895d2 · e9895d2
1 parent 05b183f
commit e9895d2
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/gguf-py/gguf/gguf_writer.py b/gguf-py/gguf/gguf_writer.py
@@ -358,8 +358,6 @@ def write_tensor_data(self, tensor: np.ndarray[Any, Any]) -> None:
         # TODO: cleaner way to get the first key
         first_tensor_name = [name for name, _ in zip(self.tensors[file_id].keys(), range(1))][0]
         ti = self.tensors[file_id].pop(first_tensor_name)
-        assert len(ti.shape) == len(tensor.shape)
-        assert all(dim1 == dim2 for dim1, dim2 in zip(ti.shape, tensor.shape))
         assert ti.nbytes == tensor.nbytes
 
         self.write_padding(fout, fout.tell())