Merge branch 'llama32-TG' of https://github.com/tenstorrent/tt-metal …

…into llama32-TG
tenstorrent · Dec 10, 2024 · f05f253 · f05f253
2 parents 886df26 + 108b8a8
commit f05f253
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/models/demos/llama3/tests/test_llama_model_prefill.py b/models/demos/llama3/tests/test_llama_model_prefill.py
@@ -186,7 +186,6 @@ def test_llama_model_inference(
 
     tt_prefill_input = model_args.prepare_residual_tensor_prefill(
         pt_prefill_input,
-        force_replicated=False if model_args.is_galaxy else True,
     )
     for i in range(1):
         start_pos = 0

diff --git a/models/demos/llama3/tt/distributed_norm.py b/models/demos/llama3/tt/distributed_norm.py
@@ -8,7 +8,7 @@
 
 
 class DistributedNorm(LightweightModule):
-    def __init__(self, norm, args, TG):
+    def __init__(self, norm, args, TG=False):
         self.norm = norm
         self.args = args