We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 6986529 commit d8830edCopy full SHA for d8830ed
swift/tuners/lora_layers.py
@@ -264,7 +264,8 @@ def inject_adapter(self,
264
model: nn.Module,
265
adapter_name: str,
266
autocast_adapter_dtype: bool = True,
267
- low_cpu_mem_usage: bool = False):
+ low_cpu_mem_usage: bool = False,
268
+ **kwargs):
269
r"""
270
Override code:
271
1. ModulesToSaveWrapper construction method: add module_key=key argument to offload to cpu
0 commit comments