From: Xuan-Son Nguyen <redacted>
Date: Thu, 3 Jul 2025 08:03:06 +0000 (+0200)
Subject: convert : correct gemma 3n conversion (#14450)
X-Git-Tag: upstream/0.0.5882~64
X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=0c2ee38ab729f3f6a7c28dc4c5dcd8d5f0cbf8fc;p=pkg%2Fggml%2Fsources%2Fllama.cpp

convert : correct gemma 3n conversion (#14450)

* convert : correct gemma 3n conversion

* rm redundant code
---

diff --git a/gguf-py/gguf/gguf_writer.py b/gguf-py/gguf/gguf_writer.py
index 697e057c..a7ecf3d3 100644
--- a/gguf-py/gguf/gguf_writer.py
+++ b/gguf-py/gguf/gguf_writer.py
@@ -714,8 +714,8 @@ class GGUFWriter:
     def add_clamp_kqv(self, value: float) -> None:
         self.add_float32(Keys.Attention.CLAMP_KQV.format(arch=self.arch), value)
 
-    def add_shared_kv_layers(self, value: float) -> None:
-        self.add_float32(Keys.Attention.SHARED_KV_LAYERS.format(arch=self.arch), value)
+    def add_shared_kv_layers(self, value: int) -> None:
+        self.add_uint32(Keys.Attention.SHARED_KV_LAYERS.format(arch=self.arch), value)
 
     def add_sliding_window_pattern(self, value: Sequence[bool]) -> None:
         self.add_array(Keys.Attention.SLIDING_WINDOW_PATTERN.format(arch=self.arch), value)