Skip to content

GitLab

"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "e54a1b49aa6268c484625c6374f952f318914743"
Unverified Commit 0e4f7270 authored by Lysandre Debut's avatar Lysandre Debut Committed by GitHub
Browse files

Remove side effects of disabling gradient computaiton (#13257)

parent b1198a84
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
src/transformers/onnx/convert.py
View file @ 0e4f7270
...@@ -127,6 +127,7 @@ def export( ...@@ -127,6 +127,7 @@ def export(
) )
config.restore_ops() config.restore_ops()
torch.set_grad_enabled(False)
return matched_inputs, onnx_outputs return matched_inputs, onnx_outputs
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment