vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

convert-hf : make model class definitions self-contained (#5825)

Browse source
This commit is contained in:
Jared Van Bortel 2024-03-02 12:21:47 -05:00 • committed by GitHub
parent bbde6eb256
commit c7a0ad8ec9
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 101 additions and 105 deletions
Download patch file Download diff file Expand all files Collapse all files

2
gguf-py/gguf/gguf_writer.py
View file

@ -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)