Merge branch 'master' into pr/7359

This commit is contained in:
Georgi Gerganov 2024-07-04 16:41:27 +03:00
commit 22a648f8cc
No known key found for this signature in database
GPG key ID: 449E073F9DC10735
88 changed files with 1892 additions and 1441 deletions

View file

@ -561,6 +561,9 @@ class GGUFWriter:
def add_relative_attn_buckets_count(self, value: int) -> None:
self.add_uint32(Keys.Attention.REL_BUCKETS_COUNT.format(arch=self.arch), value)
def add_sliding_window(self, value: int) -> None:
self.add_uint32(Keys.Attention.SLIDING_WINDOW.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.value)