From 6a05183b97992d63ccb396249caba54a981f3497 Mon Sep 17 00:00:00 2001 From: Christian Zhou-Zheng <59622928+christianazinn@users.noreply.github.com> Date: Thu, 6 Jun 2024 08:28:10 -0400 Subject: [PATCH] GGUFWriter compatibility fix Co-authored-by: compilade --- gguf-py/gguf/gguf_manager.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/gguf-py/gguf/gguf_manager.py b/gguf-py/gguf/gguf_manager.py index 2fcaf3edf..5d6133fe6 100644 --- a/gguf-py/gguf/gguf_manager.py +++ b/gguf-py/gguf/gguf_manager.py @@ -155,7 +155,7 @@ class GGUFManager(GGUFWriter): self.state = WriterState.KV_DATA - def write_tensors_to_file(self, progress: bool = False) -> None: + def write_tensors_to_file(self, *, progress: bool = False) -> None: if self.state is not WriterState.KV_DATA: raise ValueError(f'Expected GGUFManager state to be KV_DATA, got {self.state}')