]> git.djapps.eu Git - pkg/ggml/sources/llama.cpp/commitdiff
convert : filter lora tensor names (#23077)
authorSigbjørn Skjæret <redacted>
Tue, 19 May 2026 06:44:25 +0000 (08:44 +0200)
committerGitHub <redacted>
Tue, 19 May 2026 06:44:25 +0000 (09:44 +0300)
convert_lora_to_gguf.py

index 1b7334617d107e95e13ebcade34debf42884d39c..81658ba03d81e6f34af1bec6e3f92ac19fe9f29e 100755 (executable)
@@ -445,6 +445,11 @@ if __name__ == '__main__':
                     if self.lazy:
                         tensor = LazyTorchTensor.from_eager(tensor)
                     base_name = get_base_tensor_name(name)
+                    # filter base name, ignore tensor transformations for now
+                    data_gen = lambda g=tensor: g  # noqa: E731
+                    if (titem := self.filter_tensors((base_name, data_gen))) is None:
+                        continue
+                    base_name, _ = titem
                     # note: mergekit-extract-lora also adds token embeddings to the adapter
                     is_lora_a = ".lora_A.weight" in name or ".lora_embedding_A" in name
                     is_lora_b = ".lora_B.weight" in name or ".lora_embedding_B" in name