Remove unload

huchenlei · huchenlei · commit 66eeb4e1b4d3 · 2024-12-31T23:29:08.000-05:00
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):
