riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

Remove side effects of disabling gradient computaiton (#13257)

Browse Source
This commit is contained in:
Lysandre Debut 2021-08-25 11:32:51 +02:00 committed by GitHub
parent b1198a8440
commit 0e4f727069
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/transformers/onnx/convert.py
View File

@ -127,6 +127,7 @@ def export(
)
config.restore_ops()
torch.set_grad_enabled(False)
return matched_inputs, onnx_outputs