From 120a986cf0cd29812d2f50604d5957eaebdedc06 Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Fri, 16 Aug 2024 19:15:28 +0300 Subject: [PATCH] Update train_util.py --- library/train_util.py | 1 - 1 file changed, 1 deletion(-) 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