From 43be770ac7c3f1b277794ce98d2eff7b42c50718 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Laure=CE=B7t?= Date: Thu, 15 Feb 2024 15:06:26 +0100 Subject: [PATCH] Update src/refiners/foundationals/latent_diffusion/style_aligned.py MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Co-authored-by: Cédric Deltheil <355031+deltheil@users.noreply.github.com> --- src/refiners/foundationals/latent_diffusion/style_aligned.py | 1 - 1 file changed, 1 deletion(-) diff --git a/src/refiners/foundationals/latent_diffusion/style_aligned.py b/src/refiners/foundationals/latent_diffusion/style_aligned.py index afcf23a80..09ec4d00b 100644 --- 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,