Update train_util.py

2024-08-16 19:15:28 +03:00
parent 17e6274fe8
commit 120a986cf0
1 changed files with 0 additions and 1 deletions
@@ -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