Update gguf-py/gguf/gguf_writer.py

Co-authored-by: compilade <git@compilade.net>
This commit is contained in:
Younes Belkada 2024-08-18 18:41:42 +04:00 committed by GitHub
parent 9e22bb7ed6
commit 4553502d4a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -715,7 +715,7 @@ class GGUFWriter:
def add_rope_scaling_finetuned(self, value: bool) -> None: def add_rope_scaling_finetuned(self, value: bool) -> None:
self.add_bool(Keys.Rope.SCALING_FINETUNED.format(arch=self.arch), value) self.add_bool(Keys.Rope.SCALING_FINETUNED.format(arch=self.arch), value)
def add_mamba_dt_b_c_rms(self, value: bool) -> None: def add_ssm_dt_b_c_rms(self, value: bool) -> None:
self.add_bool(Keys.SSM.DT_B_C_RMS.format(arch=self.arch), value) self.add_bool(Keys.SSM.DT_B_C_RMS.format(arch=self.arch), value)
def add_rope_scaling_yarn_log_mul(self, value: float) -> None: def add_rope_scaling_yarn_log_mul(self, value: float) -> None: