Update src/refiners/foundationals/latent_diffusion/style_aligned.py

Co-authored-by: Cédric Deltheil <[email protected]>
finegrain-ai · Feb 15, 2024 · 43be770 · 43be770
1 parent 8a744fc
commit 43be770
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/src/refiners/foundationals/latent_diffusion/style_aligned.py b/src/refiners/foundationals/latent_diffusion/style_aligned.py
@@ -262,7 +262,6 @@ def style_aligned_layers(self) -> fl.Distribute:
         return fl.Distribute(*self._style_aligned_layers)
 
     def inject(self, parent: fl.Chain | None = None) -> "SharedSelfAttentionAdapter":
-        # insert the StyleAligned modules, right after the Q, K, V Linear-s (wrapped inside a fl.Distribute)
         self.target.insert_before_type(
             module_type=fl.ScaledDotProductAttention,
             new_module=self.style_aligned_layers,