diff --git a/src/nanotron/constants.py b/src/nanotron/constants.py index 580bd99d..81b78e20 100644 --- a/src/nanotron/constants.py +++ b/src/nanotron/constants.py @@ -6,7 +6,7 @@ PY_VERSION = parse(platform.python_version()) -#### FOR SERIALIZATION #### +##### FOR SERIALIZATION ##### CHECKPOINT_FILE_NAME = "checkpoint_metadata.json" MODEL_CONFIG_FILE_NAME = "model_config.json"