diff options
| author | Jared Van Bortel <jared@nomic.ai> | 2024-03-02 12:21:47 -0500 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-03-02 12:21:47 -0500 |
| commit | c7a0ad8ec9ebb5ddb1c1c80c82f2ee041c525d47 (patch) | |
| tree | 221fe201f4be6f999a8cb5ed064338e050410dbb /gguf-py | |
| parent | bbde6eb2561153aabbdfac5001c690fe00cad639 (diff) | |
convert-hf : make model class definitions self-contained (#5825)
Diffstat (limited to 'gguf-py')
| -rw-r--r-- | gguf-py/gguf/gguf_writer.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/gguf-py/gguf/gguf_writer.py b/gguf-py/gguf/gguf_writer.py index e4681475..80116083 100644 --- a/gguf-py/gguf/gguf_writer.py +++ b/gguf-py/gguf/gguf_writer.py @@ -362,7 +362,7 @@ class GGUFWriter: self.add_bool(Keys.Attention.CAUSAL.format(arch=self.arch), value) def add_pooling_type(self, value: PoolingType) -> None: - self.add_uint32(Keys.LLM.POOLING_TYPE.format(arch=self.arch), value) + self.add_uint32(Keys.LLM.POOLING_TYPE.format(arch=self.arch), value.value) def add_rope_dimension_count(self, count: int) -> None: self.add_uint32(Keys.Rope.DIMENSION_COUNT.format(arch=self.arch), count) |