nit

Author: huchenlei

lib_layerdiffusion/attention_sharing.py

@@ -346,7 +346,7 @@ def unload_model_clones(model, unload_weights_only=True, force_unload=True):
         unload_weight = True
 
     if not force_unload:
-        if unload_weights_only and unload_weight == False:
+        if unload_weights_only and unload_weight is False:
             return None
     else:
         unload_weight = True
@@ -356,6 +356,7 @@ def unload_model_clones(model, unload_weights_only=True, force_unload=True):
 
     return unload_weight
 
+
 class AttentionSharingPatcher(torch.nn.Module):
     def __init__(self, unet, frames=2, use_control=True, rank=256):
         super().__init__()