Skip to content

Commit 0c2ee38

Browse files
authored
convert : correct gemma 3n conversion (#14450)
* convert : correct gemma 3n conversion * rm redundant code
1 parent a70c8a0 commit 0c2ee38

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

gguf-py/gguf/gguf_writer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -714,8 +714,8 @@ def add_max_alibi_bias(self, bias: float) -> None:
714714
def add_clamp_kqv(self, value: float) -> None:
715715
self.add_float32(Keys.Attention.CLAMP_KQV.format(arch=self.arch), value)
716716

717-
def add_shared_kv_layers(self, value: float) -> None:
718-
self.add_float32(Keys.Attention.SHARED_KV_LAYERS.format(arch=self.arch), value)
717+
def add_shared_kv_layers(self, value: int) -> None:
718+
self.add_uint32(Keys.Attention.SHARED_KV_LAYERS.format(arch=self.arch), value)
719719

720720
def add_sliding_window_pattern(self, value: Sequence[bool]) -> None:
721721
self.add_array(Keys.Attention.SLIDING_WINDOW_PATTERN.format(arch=self.arch), value)

0 commit comments

Comments
 (0)