Remove prints

injection_utils.py

@@ -129,18 +129,7 @@ def regiter_attention_editor_diffusers(model, editor: AttentionBase):
 
         return forward
 
-    #def conv_forward(self):
-    #    def forward(x):
-    #        return editor.conv(self, x)
-
-    #    return forward
-
     def register_editor(net, count, place_in_unet, prefix=''):
-        #print(prefix + f'Found net: {net.__class__.__name__}')
-        #if 'Conv' in net.__class__.__name__:
-        #    print(prefix + f'Found conv with kernel size: {net.kernel_size}')
-        #    net.old_forward = net.forward
-        #    net.forward = conv_forward(net)
         for name, subnet in net.named_children():
             if net.__class__.__name__ == 'Attention':  # spatial Transformer layer
                 net.forward = ca_forward(net, place_in_unet)
@@ -153,14 +142,10 @@ def regiter_attention_editor_diffusers(model, editor: AttentionBase):
     for net_name, net in model.unet.named_children():
         if "down" in net_name:
             cross_att_count += register_editor(net, 0, "down")
-            #print(f'Down number of attention layers {cross_att_count}!')
         elif "mid" in net_name:
             cross_att_count += register_editor(net, 0, "mid")
-            #print(f'Mid number of attention layers {cross_att_count}!')
         elif "up" in net_name:
             cross_att_count += register_editor(net, 0, "up")
-            #print(f'Up number of attention layers {cross_att_count}!')
-    print(f'Number of attention layers {cross_att_count}!')
     editor.num_att_layers = cross_att_count
     editor.model = model
     model.editor = editor