Skip to content

Commit f564e02

Browse files
committed
correct norm name & condition
1 parent 11e0c73 commit f564e02

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

convert_lora_to_gguf.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -389,7 +389,7 @@ def get_tensors(self) -> Iterator[tuple[str, Tensor]]:
389389
if ".base_layer.weight" in name:
390390
continue
391391
# mergekit-extract-lora add these layernorm to the adapter, we need to keep them
392-
if ".layernorm" or ".norm" in name:
392+
if "_layernorm" in name or ".norm" in name:
393393
yield (base_name, tensor)
394394
continue
395395
logger.error(f"Unexpected name '{name}': Not a lora_A or lora_B tensor")

0 commit comments

Comments
 (0)