Revert "#0: Update reference model"

This reverts commit bd491a2.
tenstorrent · Feb 5, 2025 · d650d26 · d650d26
1 parent 146f1d8
commit d650d26
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/models/demos/llama3/tt/llama_attention.py b/models/demos/llama3/tt/llama_attention.py
@@ -136,7 +136,7 @@ def __init__(
                 dtype=self.dtype,
                 memory_config=ttnn.DRAM_MEMORY_CONFIG,
                 layout=ttnn.TILE_LAYOUT,
-                # cache_file_name=cache_name("wqkv_bias_sharded"),
+                cache_file_name=cache_name("wqkv_bias_sharded"),
             )
             self.wqkv_bias_prefill = ttnn.reshape(self.wqkv_bias, ttnn.Shape([1, 1, 1, self.wqkv_bias.shape[-1]]))