Skip to content

Commit ff7e6af

Browse files
committed
Fix(peft): Re-apply group offloading after deleting adapters
1 parent 0c75892 commit ff7e6af

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

src/diffusers/loaders/peft.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,7 @@
2222
import safetensors
2323
import torch
2424

25+
from ..hooks.group_offloading import _maybe_remove_and_reapply_group_offloading
2526
from ..utils import (
2627
MIN_PEFT_VERSION,
2728
USE_PEFT_BACKEND,
@@ -792,6 +793,8 @@ def delete_adapters(self, adapter_names: Union[List[str], str]):
792793
if hasattr(self, "peft_config"):
793794
self.peft_config.pop(adapter_name, None)
794795

796+
_maybe_remove_and_reapply_group_offloading(self)
797+
795798
def enable_lora_hotswap(
796799
self, target_rank: int = 128, check_compiled: Literal["error", "warn", "ignore"] = "error"
797800
) -> None:

0 commit comments

Comments
 (0)