We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent be211c8 commit e0b7817Copy full SHA for e0b7817
torchtitan/components/ft/diloco/utils.py
@@ -6,7 +6,7 @@
6
7
import torch.nn as nn
8
from torchtitan.components.ft.config import FaultTolerance as FTConfig
9
-from torchtitan.distributed.pipeline import generate_llm_fqn_per_model_part
+from torchtitan.distributed.pipeline_parallel import generate_llm_fqn_per_model_part
10
11
12
def module_split(
0 commit comments