Skip to content

Commit 019b7ff

Browse files
fix adalora device mismatch issue (#583)
1 parent b519e3f commit 019b7ff

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/peft/tuners/lora.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -336,6 +336,8 @@ def _replace_module(self, parent_module, child_name, new_module, old_module):
336336
for name, module in new_module.named_modules():
337337
if "lora_" in name:
338338
module.to(old_module.weight.device)
339+
if "ranknum" in name:
340+
module.to(old_module.weight.device)
339341

340342
def __getattr__(self, name: str):
341343
"""Forward missing attributes to the wrapped module."""

0 commit comments

Comments
 (0)