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 5b5d468 commit a310420Copy full SHA for a310420
torchtitan/experiments/forge/job_config.py
@@ -12,7 +12,7 @@
12
Checkpoint,
13
Comm,
14
Compile,
15
- Float8,
+ Float8Dense,
16
LRScheduler,
17
Model,
18
Optimizer,
@@ -33,7 +33,7 @@ class ForgeJobConfig:
33
default_factory=ActivationCheckpoint
34
)
35
compile: Compile = field(default_factory=Compile)
36
- float8: Float8 = field(default_factory=Float8)
+ float8: Float8Dense = field(default_factory=Float8Dense)
37
comm: Comm = field(default_factory=Comm)
38
39
def to_dict(self) -> dict[str, Any]:
0 commit comments