Remove unload

huchenlei · Jan 1, 2025 · 66eeb4e · 66eeb4e
1 parent ab409ee
commit 66eeb4e
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/lib_layerdiffusion/attention_sharing.py b/lib_layerdiffusion/attention_sharing.py
@@ -327,7 +327,6 @@ def __init__(self, layer_list):
 class AttentionSharingPatcher(torch.nn.Module):
     def __init__(self, unet, frames=2, use_control=True, rank=256):
         super().__init__()
-        model_management.unload_model_clones(unet)
 
         units = []
         for i in range(32):