diff --git a/src/zeroband/checkpoint.py b/src/zeroband/checkpoint.py index 9d486560..f3527ad8 100644 --- a/src/zeroband/checkpoint.py +++ b/src/zeroband/checkpoint.py @@ -28,9 +28,6 @@ ## code inspired by torchtitan https://github.com/pytorch/torchtitan/blob/main/torchtitan/checkpoint.py -GLOBAL_STATE_FILE = "global_state_dict.pt" - - @dataclass class TrainingProgress(Stateful): total_tokens: int