diff --git a/library/train_util.py b/library/train_util.py index f1991cd93..2b924846e 100644 --- a/library/train_util.py +++ b/library/train_util.py @@ -3906,6 +3906,7 @@ def prepare_accelerator(args: argparse.Namespace): kwargs_handlers=kwargs_handlers, dynamo_backend=dynamo_backend, ) + print("accelerator device:", accelerator.device) return accelerator