From 800ba08f8f36c710646e8a6d4c6f36bf2c7685f3 Mon Sep 17 00:00:00 2001 From: Johan Hansson <39947546+JohanWork@users.noreply.github.com> Date: Wed, 17 Apr 2024 08:32:30 +0000 Subject: [PATCH] fix validation step --- src/axolotl/utils/config/models/input/v0_4_1/__init__.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/axolotl/utils/config/models/input/v0_4_1/__init__.py b/src/axolotl/utils/config/models/input/v0_4_1/__init__.py index 6eaf34c547..2264c38cd6 100644 --- a/src/axolotl/utils/config/models/input/v0_4_1/__init__.py +++ b/src/axolotl/utils/config/models/input/v0_4_1/__init__.py @@ -689,10 +689,10 @@ def check_saves(cls, data): @classmethod def check_push_save(cls, data): if data.get("hub_model_id") and not ( - data.get("save_steps") or data.get("saves_per_epoch") + data.get("save_strategy") ): LOG.warning( - "hub_model_id is set without any models being saved. To save a model, set either save_steps or saves_per_epoch." + "hub_model_id is set without any models being saved. To save a model, set save_strategy." ) return data