Skip to content

Commit d8830ed

Browse files
tastelikefeetJintao-Huang
authored andcommitted
fix peft 0.17 ci (#5353)
1 parent 6986529 commit d8830ed

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

swift/tuners/lora_layers.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -264,7 +264,8 @@ def inject_adapter(self,
264264
model: nn.Module,
265265
adapter_name: str,
266266
autocast_adapter_dtype: bool = True,
267-
low_cpu_mem_usage: bool = False):
267+
low_cpu_mem_usage: bool = False,
268+
**kwargs):
268269
r"""
269270
Override code:
270271
1. ModulesToSaveWrapper construction method: add module_key=key argument to offload to cpu

0 commit comments

Comments
 (0)