Merge pull request #1221 from clayne/1734205326-gradient-checkpointin…

…g-interval-error-fix safety_check: Fix gradient checkpointing interval error message
bghira · Dec 14, 2024 · 2d52cda · 2d52cda
2 parents b323d1b + ad47a58
commit 2d52cda
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/helpers/training/default_settings/safety_check.py b/helpers/training/default_settings/safety_check.py
@@ -150,7 +150,7 @@ def safety_check(args, accelerator):
             not in gradient_checkpointing_interval_supported_models
         ):
             logger.error(
-                f"Gradient checkpointing is not supported with {args.model_family} models. Please disable --gradient_checkpointing_interval by setting it to None, or remove it from your configuration. Currently supported models: {gradient_checkpointing_interval_supported_models}"
+                f"Gradient checkpointing interval is not supported with {args.model_family} models. Please disable --gradient_checkpointing_interval by setting it to None, or remove it from your configuration. Currently supported models: {gradient_checkpointing_interval_supported_models}"
             )
             sys.exit(1)
         if args.gradient_checkpointing_interval == 0: