#13070: fix SD (#13073)

tenstorrent · Sep 24, 2024 · c4bb801 · c4bb801
1 parent 1dbdc62
commit c4bb801
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/models/demos/wormhole/stable_diffusion/tt/ttnn_functional_cross_attention.py b/models/demos/wormhole/stable_diffusion/tt/ttnn_functional_cross_attention.py
@@ -407,7 +407,7 @@ def time_sharded_attention(self, query, t_key, value, head_size, attn_type):
                 v_slice = ttnn.slice(
                     value,
                     (j, i, 0, 0),
-                    (j, i, self.seq_len, self.key_len),
+                    (j + 1, i + 1, self.seq_len, self.key_len),
                     memory_config=self.l1_interleaved_memory_config,
                 )
                 mm_slice = ttnn.matmul(