stylestudio

ip_adapter/attention_processor.py

@@ -866,7 +866,8 @@ class AttnProcessor2_0_hijack(torch.nn.Module):
         hidden_states = hidden_states / attn.rescale_output_factor
 
         if self.denoise_step == self.num_inference_step:
-            self.denoise_step == 0
+            self.denoise_step = 0
+            
         return hidden_states
     
 class IPAttnProcessor2_0_cross_modal(torch.nn.Module):
@@ -1031,6 +1032,6 @@ class IPAttnProcessor2_0_cross_modal(torch.nn.Module):
         hidden_states = hidden_states / attn.rescale_output_factor
 
         if self.denoise_step == self.num_inference_step:
-            self.denoise_step == 0
+            self.denoise_step = 0
 
         return hidden_states

ip_adapter/ip_adapter.py

@@ -965,9 +965,7 @@ class StyleStudio_Adapter(CSGO):
             if cross_attention_dim is None:
                 attn_procs[name] = AttnProcessor_hijack(
                                         fuSAttn=self.fuSAttn,
-                                        fuScale=self.fuScale,
-                                        end_fusion=self.end_fusion,
-                                        attn_name=name)
+                                        end_fusion=self.end_fusion,)
             else:
                 # layername_id += 1
                 selected = False
@@ -984,9 +982,7 @@ class StyleStudio_Adapter(CSGO):
                             fuAttn=self.fuAttn,
                             fuIPAttn=self.fuIPAttn,
                             adainIP=self.adainIP,
-                            fuScale=self.fuScale,
                             end_fusion=self.end_fusion,
-                            attn_name=name,
                         )
                 if selected is False:
                     attn_procs[name] = IPAttnProcessor_cross_modal(
@@ -995,12 +991,9 @@ class StyleStudio_Adapter(CSGO):
                         num_style_tokens=self.num_style_tokens,
                         skip=True,
                         fuAttn=self.fuAttn,
-
                         fuIPAttn=self.fuIPAttn,
                         adainIP=self.adainIP,
-                        fuScale=self.fuScale,
                         end_fusion=self.end_fusion,
-                        attn_name=name,
                     )
 
                 attn_procs[name].to(self.device, dtype=torch.float16)