From c12cbfced3059dd2d80230ced7f8db0cfab71ae4 Mon Sep 17 00:00:00 2001 From: Maw-Fox Date: Fri, 11 Nov 2022 17:43:09 -0700 Subject: [PATCH] Fixed ref typo. --- trainer/diffusers_trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/trainer/diffusers_trainer.py b/trainer/diffusers_trainer.py index 34adf60..2227d83 100644 --- a/trainer/diffusers_trainer.py +++ b/trainer/diffusers_trainer.py @@ -733,7 +733,7 @@ def main(): store.validator.completed() - if args.resize and args.migration: + if args.resize and args.data_migration: print(f"Completed resize and migration to '{args.dataset}_cropped' please relaunch the trainer without the --resize argument and train on the migrated dataset.") exit(0)