Skip to content

Commit 029314a

Browse files
Implementing review changes
1 parent 7f52f2f commit 029314a

File tree

1 file changed

+5
-6
lines changed

1 file changed

+5
-6
lines changed

gguf-py/gguf/scripts/gguf_editor_gui.py

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1551,18 +1551,17 @@ def save_file(self):
15511551

15521552
writer.add_key_value(key, value, value_type)
15531553

1554-
# Copy tensors
1554+
# Add tensors (including data)
15551555
for tensor in self.reader.tensors:
1556-
writer.add_tensor_info(tensor.name, tensor.data.shape, tensor.data.dtype, tensor.data.nbytes, tensor.tensor_type)
1556+
writer.add_tensor(tensor.name, tensor.data, raw_shape=tensor.data.shape, raw_dtype=tensor.tensor_type)
15571557

15581558
# Write header and metadata
1559+
writer.open_output_file(Path(file_path))
15591560
writer.write_header_to_file()
15601561
writer.write_kv_data_to_file()
1561-
writer.write_ti_data_to_file()
15621562

1563-
# Write tensor data
1564-
for tensor in self.reader.tensors:
1565-
writer.write_tensor_data(tensor.data)
1563+
# Write tensor data using the optimized method
1564+
writer.write_tensors_to_file(progress=False)
15661565

15671566
writer.close()
15681567

0 commit comments

Comments
 (0)