Skip to content

Commit

Permalink
GGUFWriter compatibility fix
Browse files Browse the repository at this point in the history
Co-authored-by: compilade <[email protected]>
  • Loading branch information
christianazinn and compilade authored Jun 6, 2024
1 parent 706bd69 commit 6a05183
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion gguf-py/gguf/gguf_manager.py
Original file line number Diff line number Diff line change
Expand Up @@ -155,7 +155,7 @@ def write_kv_data_to_file(self) -> None:

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}')

Expand Down

0 comments on commit 6a05183

Please sign in to comment.