Skip to content

Commit e0b7817

Browse files
committed
quick fix import error
1 parent be211c8 commit e0b7817

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torchtitan/components/ft/diloco/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66

77
import torch.nn as nn
88
from torchtitan.components.ft.config import FaultTolerance as FTConfig
9-
from torchtitan.distributed.pipeline import generate_llm_fqn_per_model_part
9+
from torchtitan.distributed.pipeline_parallel import generate_llm_fqn_per_model_part
1010

1111

1212
def module_split(

0 commit comments

Comments
 (0)