Skip to content

Commit 32dd0d5

Browse files
committed
fix enabling/disabling adapters
1 parent 61e316c commit 32dd0d5

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/diffusers/utils/peft_utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -163,9 +163,9 @@ def set_adapter_layers(model, enabled=True):
163163
if isinstance(module, BaseTunerLayer):
164164
# The recent version of PEFT needs to call `enable_adapters` instead
165165
if hasattr(module, "enable_adapters"):
166-
module.enable_adapters(enabled=False)
166+
module.enable_adapters(enabled=enabled)
167167
else:
168-
module.disable_adapters = True
168+
module.disable_adapters = not enabled
169169

170170

171171
def set_weights_and_activate_adapters(model, adapter_names, weights):

0 commit comments

Comments
 (0)