diff --git a/library/train_util.py b/library/train_util.py index 83efa03..b92057e 100644 --- a/library/train_util.py +++ b/library/train_util.py @@ -4740,7 +4740,6 @@ def prepare_accelerator(args: argparse.Namespace): kwargs_handlers=kwargs_handlers, dynamo_backend=dynamo_backend, deepspeed_plugin=deepspeed_plugin, - mixed_precision='bf16', ) print("accelerator device:", accelerator.device) return accelerator