Просмотр исходного кода

gguf-py : use copy-on-write mode for localtensor (#18162)

Sigbjørn Skjæret 4 недель назад
Родитель
Сommit
ec7b9329ae
1 измененных файлов с 1 добавлено и 1 удалено
  1. 1 1
      gguf-py/gguf/utility.py

+ 1 - 1
gguf-py/gguf/utility.py

@@ -288,7 +288,7 @@ class LocalTensor:
     data_range: LocalTensorRange
 
     def mmap_bytes(self) -> np.ndarray:
-        return np.memmap(self.data_range.filename, mode='r', offset=self.data_range.offset, shape=self.data_range.size)
+        return np.memmap(self.data_range.filename, mode='c', offset=self.data_range.offset, shape=self.data_range.size)
 
 
 class SafetensorsLocal: