Skip to content

Commit 45f0291

Browse files
bvandermoonajkv-google
authored andcommitted
Migrate MaxText/train.py to maxtext/trainers/pre_train/train.py
1 parent 4b515df commit 45f0291

2 files changed

Lines changed: 4 additions & 1 deletion

File tree

src/maxtext/trainers/pre_train/train.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,8 @@
1+
<<<<<<< HEAD
12
# Copyright 2023–2025 Google LLC
3+
=======
4+
# Copyright 2023–2026 Google LLC
5+
>>>>>>> 9194ee09f (Migrate MaxText/train.py to maxtext/trainers/pre_train/train.py)
26
#
37
# Licensed under the Apache License, Version 2.0 (the "License");
48
# you may not use this file except in compliance with the License.

tools/gcs_benchmarks/standalone_checkpointer.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,6 @@
3838

3939
Transformer = models.transformer_as_linen
4040

41-
4241
def checkpoint_loop(config, state=None):
4342
"""Main Checkpointing loop.
4443

0 commit comments

Comments
 (0)