diff --git a/tt-train/cmake/dependencies.cmake b/tt-train/cmake/dependencies.cmake
index 4e284f34a98..2c9811fc6a1 100644
--- a/tt-train/cmake/dependencies.cmake
+++ b/tt-train/cmake/dependencies.cmake
@@ -72,6 +72,8 @@ CPMAddPackage(
         "XTENSOR_ENABLE_TESTS OFF"
 )
 
+CPMAddPackage(NAME taskflow GITHUB_REPOSITORY taskflow/taskflow GIT_TAG v3.7.0)
+
 include(${PROJECT_SOURCE_DIR}/cmake/fetch_msgpack.cmake)
 
 include(${PROJECT_SOURCE_DIR}/cmake/fetch_cli11.cmake)
diff --git a/tt-train/sources/ttml/CMakeLists.txt b/tt-train/sources/ttml/CMakeLists.txt
index c6b8a7496c4..ed7344f8ff2 100644
--- a/tt-train/sources/ttml/CMakeLists.txt
+++ b/tt-train/sources/ttml/CMakeLists.txt
@@ -19,7 +19,6 @@ if(NOT TARGET Metalium::Metal)
         "$ENV{TT_METAL_HOME}"
         "$ENV{TT_METAL_HOME}/tt_metal"
         "$ENV{TT_METAL_HOME}/tt_metal/third_party/umd"
-        "$ENV{TT_METAL_HOME}/tt_metal/third_party/taskflow"
         "$ENV{TT_METAL_HOME}/tt_metal/third_party/tracy/public"
         "$ENV{TT_METAL_HOME}/tt_metal/hw/inc/wormhole"
         "$ENV{TT_METAL_HOME}/tt_metal/hw/inc/wormhole/wormhole_b0_defines"