Skip to content

Commit a310420

Browse files
Fix forge float8 import (#1770)
In #1701 we split out the `Float8` job config to `Float8Dense` and `Float8MoE`. This forge config was inadvertently not updated. cc @tianyu-l
1 parent 5b5d468 commit a310420

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

torchtitan/experiments/forge/job_config.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
Checkpoint,
1313
Comm,
1414
Compile,
15-
Float8,
15+
Float8Dense,
1616
LRScheduler,
1717
Model,
1818
Optimizer,
@@ -33,7 +33,7 @@ class ForgeJobConfig:
3333
default_factory=ActivationCheckpoint
3434
)
3535
compile: Compile = field(default_factory=Compile)
36-
float8: Float8 = field(default_factory=Float8)
36+
float8: Float8Dense = field(default_factory=Float8Dense)
3737
comm: Comm = field(default_factory=Comm)
3838

3939
def to_dict(self) -> dict[str, Any]:

0 commit comments

Comments
 (0)