Skip to content

Commit

Permalink
quick fix
Browse files Browse the repository at this point in the history
  • Loading branch information
eliebak committed Dec 2, 2024
1 parent cfcdeae commit 521f759
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions src/nanotron/serialize/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -105,6 +105,7 @@ def save(

save_lr_scheduler(
lr_scheduler=lr_scheduler,
is_zero=config.optimizer.zero_stage,
parallel_context=parallel_context,
root_folder=root_folder,
)
Expand Down

0 comments on commit 521f759

Please sign in to comment.