diff --git a/gguf-py/gguf/gguf_writer.py b/gguf-py/gguf/gguf_writer.py index 13f231261..85a868a77 100644 --- a/gguf-py/gguf/gguf_writer.py +++ b/gguf-py/gguf/gguf_writer.py @@ -358,8 +358,6 @@ class GGUFWriter: # 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())