Skip to content

Commit 46a803f

Browse files
committed
Ignore potential junk tensors in LoRA
1 parent 81cd6b7 commit 46a803f

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

exllamav2/lora.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -81,6 +81,8 @@ def __init__(self,
8181
f = load_file(self.lora_path, map_location = "cpu")
8282

8383
for key in f.keys():
84+
if any(key.endswith(x) for x in [".original_module.weight", ".modules_to_save.weight"]):
85+
continue
8486
tensor = f[key]
8587

8688
# Find target

0 commit comments

Comments
 (0)