fix hidiff

2026-09-20 01:31:13 +02:00 · 2024-05-17 11:21:30 -04:00
parent 7319498f57
commit beae2f88a6
1 changed files with 2 additions and 0 deletions
@@ -1666,6 +1666,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,
@@ -1676,6 +1677,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)