diff --git a/gguf.py b/gguf.py index de75b54d4..1888b6c2d 100644 --- a/gguf.py +++ b/gguf.py @@ -283,6 +283,7 @@ if __name__ == "__main__": tensor2 = np.ones((32,), dtype=np.float32) * 101.0 gguf_writer.write_tensor_info("tensor0", tensor1) gguf_writer.write_tensor_info("tensor1", tensor2) - gguf_writer.write_tensors() + gguf_writer.write_tensor(tensor1) + gguf_writer.write_tensor(tensor2) gguf_writer.close()