diff --git a/trainer/diffusers_trainer.py b/trainer/diffusers_trainer.py index d42e9af..b371919 100644 --- a/trainer/diffusers_trainer.py +++ b/trainer/diffusers_trainer.py @@ -159,8 +159,7 @@ class Validation(): def __init__(self, is_skipped: bool, is_extended: bool) -> None: if is_skipped: self.validate = self.__no_op - print("Validation: Skipped") - return + return print("Validation: Skipped") if is_extended: self.validate = self.__extended_validate