Spaces:

Westlake-AGI-Lab
/

StyleStudio

Running on Zero

Leimingkun commited on 23 days ago

Commit

123f4ac

•

1 Parent(s): c7a40ad

stylestudio

Files changed (1) hide show

ip_adapter/ip_adapter.py CHANGED Viewed

@@ -976,9 +976,7 @@ class StyleStudio_Adapter(CSGO):
                         attn_procs[name] = IPAttnProcessor_cross_modal(
                             hidden_size=hidden_size,
                             cross_attention_dim=cross_attention_dim,
-                            style_scale=1.0,
-                            style=True,
-                            num_style_tokens=self.num_style_tokens,
                             fuAttn=self.fuAttn,
                             fuIPAttn=self.fuIPAttn,
                             adainIP=self.adainIP,
@@ -988,7 +986,7 @@ class StyleStudio_Adapter(CSGO):
                     attn_procs[name] = IPAttnProcessor_cross_modal(
                         hidden_size=hidden_size,
                         cross_attention_dim=cross_attention_dim,
-                        num_style_tokens=self.num_style_tokens,
                         skip=True,
                         fuAttn=self.fuAttn,
                         fuIPAttn=self.fuIPAttn,

                         attn_procs[name] = IPAttnProcessor_cross_modal(
                             hidden_size=hidden_size,
                             cross_attention_dim=cross_attention_dim,
+                            num_tokens=self.num_style_tokens,
                             fuAttn=self.fuAttn,
                             fuIPAttn=self.fuIPAttn,
                             adainIP=self.adainIP,
                     attn_procs[name] = IPAttnProcessor_cross_modal(
                         hidden_size=hidden_size,
                         cross_attention_dim=cross_attention_dim,
+                        num_tokens=self.num_style_tokens,
                         skip=True,
                         fuAttn=self.fuAttn,
                         fuIPAttn=self.fuIPAttn,