manual pre-commit

tenstorrent · Mar 6, 2025 · 9a6c50d · 9a6c50d
1 parent 9f217dc
commit 9a6c50d
Show file tree

Hide file tree

Showing 9 changed files with 453 additions and 492 deletions.
diff --git a/models/demos/falcon7b_common/tt/model_utils.py b/models/demos/falcon7b_common/tt/model_utils.py
@@ -50,7 +50,9 @@ def preprocess_weights(weights_to_cache):
             layout=tt_layout,
             device=mesh_device,
             memory_config=model_config[f"{weight_config_str}_MEMCFG"],
-            mesh_mapper=ttnn.replicate_tensor_to_mesh_mapper(mesh_device) if type(mesh_device) == ttnn.MeshDevice else None,
+            mesh_mapper=ttnn.replicate_tensor_to_mesh_mapper(mesh_device)
+            if type(mesh_device) == ttnn.MeshDevice
+            else None,
             cache_file_name=str(path),
             preprocess=preprocess_weights,
         )

diff --git a/models/demos/llama3/tt/llama_common.py b/models/demos/llama3/tt/llama_common.py
@@ -402,7 +402,9 @@ def sample_host(tt_input, mesh_device, temperature=0.6, top_p=0.08, on_host=True
                 layout=ttnn.ROW_MAJOR_LAYOUT,
                 dtype=ttnn.uint32,
                 device=None,
-                mesh_mapper=ttnn.replicate_tensor_to_mesh_mapper(mesh_device) if mesh_device.get_num_devices() > 1 else None,
+                mesh_mapper=ttnn.replicate_tensor_to_mesh_mapper(mesh_device)
+                if mesh_device.get_num_devices() > 1
+                else None,
             ),
             pt_out,
         )

diff --git a/models/demos/t3000/llama2_70b/tt/llama_common.py b/models/demos/t3000/llama2_70b/tt/llama_common.py
@@ -31,6 +31,7 @@
     MeshToTensor,
 )
 
+
 class ConcatMesh2DToTensor(MeshToTensor):
     def __init__(self, mesh_device, dims, cluster_shape):
         self.dims = dims