From 3444d981eb26c42b9fa56998e7e492d0aacafb30 Mon Sep 17 00:00:00 2001 From: Tushar Jain Date: Wed, 6 Aug 2025 00:22:43 -0700 Subject: [PATCH] fix module import --- torchtitan/components/ft/diloco/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchtitan/components/ft/diloco/utils.py b/torchtitan/components/ft/diloco/utils.py index f7eaf26593..4a65ed78e4 100644 --- a/torchtitan/components/ft/diloco/utils.py +++ b/torchtitan/components/ft/diloco/utils.py @@ -6,7 +6,7 @@ import torch.nn as nn from torchtitan.components.ft.config import FaultTolerance as FTConfig -from torchtitan.distributed.pipeline import generate_llm_fqn_per_model_part +from torchtitan.distributed.pipeline_parallel import generate_llm_fqn_per_model_part def module_split(