Prepare for v0.54 release of ttnn and tt-metalium

tenstorrent · Jan 9, 2025 · 50120a2 · 50120a2
1 parent cc37ee1
commit 50120a2
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/update_tags.py b/update_tags.py
@@ -11,10 +11,13 @@
 
 if project in ["ttnn", "tt-metalium"]:
     version_no_v = version.replace("v", "")
+    #
+    # From version 0.54 forward, the wheel files were renamed to ttnn from metal_libs.
+    #
     additional_cmd = \
-        f"wget https://github.com/tenstorrent/tt-metal/releases/download/{version}/metal_libs-{version_no_v}+wormhole.b0-cp38-cp38-linux_x86_64.whl\n"
+        f"wget https://github.com/tenstorrent/tt-metal/releases/download/{version}/ttnn-{version_no_v}+wormhole.b0-cp38-cp38-linux_x86_64.whl\n"
     additional_cmd += \
-        f"pip install --extra-index-url https://download.pytorch.org/whl/cpu metal_libs-{version_no_v}+wormhole.b0-cp38-cp38-linux_x86_64.whl --force-reinstall"
+        f"pip install --extra-index-url https://download.pytorch.org/whl/cpu ttnn-{version_no_v}+wormhole.b0-cp38-cp38-linux_x86_64.whl --force-reinstall"
     version_structure = {version: {"additional_cmd": additional_cmd}}
 
 versions[project]["versions"].update(version_structure)