shahondin1624/llama.cpp
1
0
Fork 0
Code Issues 2 Pull Requests Actions 3 Packages Projects Releases Wiki Activity

convert : filter lora tensor names (#23077)

Browse Source
This commit is contained in:
Sigbjørn Skjæret
2026-05-19 08:44:25 +02:00
committed by GitHub
parent 439f1b193d
commit f1c1c5c057
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
convert_lora_to_gguf.py
+5
View File
@@ -445,6 +445,11 @@ if __name__ == '__main__':
if self.lazy: if self.lazy:
tensor = LazyTorchTensor.from_eager(tensor) tensor = LazyTorchTensor.from_eager(tensor)
base_name = get_base_tensor_name(name) 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 # 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_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 is_lora_b = ".lora_B.weight" in name or ".lora_embedding_B" in name