Replace torch.set_grad_enabled by torch.no_grad (#13703)

src/transformers/onnx/convert.py

@@ -90,7 +90,7 @@ def export(
         raise AssertionError(f"Unsupported PyTorch version, minimum required is 1.8.0, got: {torch_version}")
 
     logger.info(f"Using framework PyTorch: {torch.__version__}")
-    torch.set_grad_enabled(False)
+    with torch.no_grad():
         model.config.return_dict = True
         model.eval()
 
@@ -127,7 +127,6 @@ def export(
         )
 
         config.restore_ops()
-    torch.set_grad_enabled(True)
 
     return matched_inputs, onnx_outputs