Skip to content

Commit 47e0fb2

Browse files
committed
gguf : free tensors as they are written
1 parent 199724f commit 47e0fb2

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

gguf-py/gguf/gguf.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -661,7 +661,11 @@ def write_tensors_to_file(self):
661661

662662
self.write_padding(self.fout, self.fout.tell())
663663

664-
for tensor in self.tensors:
664+
while True:
665+
try:
666+
tensor = self.tensors.pop(0)
667+
except IndexError:
668+
break
665669
tensor.tofile(self.fout)
666670
self.write_padding(self.fout, tensor.nbytes)
667671

0 commit comments

Comments
 (0)