#0: Fix CI: Don't cache weights for quick, AllPostCommit Llama test

tenstorrent · Dec 6, 2024 · b16d446 · b16d446
1 parent b756309
commit b16d446
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/models/demos/llama3/tt/llama_embedding.py b/models/demos/llama3/tt/llama_embedding.py
@@ -23,7 +23,7 @@ def __init__(
 
         base_name = args.get_state_dict_prefix("", None) + "tok_embeddings.weight"
         torch_weight = self.state_dict[base_name].unsqueeze(0).unsqueeze(0)
-        cache_name = weight_cache_path / base_name
+        cache_name = None if args.dummy_weights else weight_cache_path / base_name
         self.weights = ttnn.as_tensor(
             torch_weight,
             dtype=dtype,