Fix test for torchneuroncore in Trainer (#22028)

a5392ee7 · Sylvain Gugger · GitHub · de81adf9 · a5392ee7
Unverified Commit a5392ee7 authored Mar 08, 2023 by Sylvain Gugger Committed by GitHub Mar 08, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/transformers/trainer.py src/transformers/trainer.py +1 -1

No files found.
--- a/src/transformers/trainer.py
+++ b/src/transformers/trainer.py
@@ -1538,7 +1538,7 @@ class Trainer:

            if self.args.ddp_bucket_cap_mb is not None:
                kwargs["bucket_cap_mb"] = self.args.ddp_bucket_cap_mb
-            if is_torch_neuroncore_available:
+            if is_torch_neuroncore_available():
                return model
            model = nn.parallel.DistributedDataParallel(
                model,