Fix group norm L1 OOM error in Stable Diffusion

tenstorrent · Dec 5, 2024 · 35b20db · 35b20db
1 parent 6996c9a
commit 35b20db
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/models/demos/wormhole/stable_diffusion/tt/ttnn_functional_resnetblock2d_new_conv.py b/models/demos/wormhole/stable_diffusion/tt/ttnn_functional_resnetblock2d_new_conv.py
@@ -416,6 +416,8 @@ def __call__(
         hidden_states = ttnn.reshape(
             hidden_states, (self.batch_size, 1, self.conv2_input_height * self.conv2_input_width, in_channels)
         )
+        hidden_states = ttnn.reallocate(hidden_states)
+
         hidden_states = ttnn.group_norm(
             hidden_states,
             num_groups=groups,