fix hidiffusion with freeu

2026-09-10 06:48:43 +02:00 · 2024-05-12 12:31:04 -04:00
parent 62edb14111
commit ffe53e4a79
1 changed files with 2 additions and 0 deletions
@@ -1542,6 +1542,7 @@ def make_diffusers_cross_attn_up_block(block_class: Type[torch.nn.Module]) -> Ty
                res_hidden_states_tuple = res_hidden_states_tuple[:-1]

                # FreeU: Only operate on the first two stages
+                """ # TODO disable hidiffusion usage of freeu
                if is_freeu_enabled:
                    hidden_states, res_hidden_states = apply_freeu(
                        self.resolution_idx,
@@ -1552,6 +1553,7 @@ def make_diffusers_cross_attn_up_block(block_class: Type[torch.nn.Module]) -> Ty
                        b1=self.b1,
                        b2=self.b2,
                    )
+                """

                hidden_states = torch.cat([hidden_states, res_hidden_states], dim=1)