(model, enabled=True)
| 210 | |
| 211 | |
| 212 | def set_adapter_layers(model, enabled=True): |
| 213 | from peft.tuners.tuners_utils import BaseTunerLayer |
| 214 | |
| 215 | for module in model.modules(): |
| 216 | if isinstance(module, BaseTunerLayer): |
| 217 | # The recent version of PEFT needs to call `enable_adapters` instead |
| 218 | if hasattr(module, "enable_adapters"): |
| 219 | module.enable_adapters(enabled=enabled) |
| 220 | else: |
| 221 | module.disable_adapters = not enabled |
| 222 | |
| 223 | |
| 224 | def delete_adapter_layers(model, adapter_name): |
no test coverage detected
searching dependent graphs…