clean code

huggingface · Sep 14, 2023 · bd5595f · bd5595f
1 parent 5b5971e
commit bd5595f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/transformers/trainer.py b/src/transformers/trainer.py
@@ -585,7 +585,7 @@ def __init__(
             if args.half_precision_backend == "cpu_amp":
                 self.use_cpu_amp = True
                 self.amp_dtype = torch.bfloat16
-            if args.half_precision_backend == "apex":
+            elif args.half_precision_backend == "apex":
                 if not is_apex_available():
                     raise ImportError(
                         "Using FP16 with APEX but APEX is not installed, please refer to"