mirror of
https://github.com/huggingface/transformers.git
synced 2025-07-31 02:02:21 +06:00
Grad enabled typo
This commit is contained in:
parent
0e4f727069
commit
46554fc12f
@ -127,7 +127,7 @@ def export(
|
||||
)
|
||||
|
||||
config.restore_ops()
|
||||
torch.set_grad_enabled(False)
|
||||
torch.set_grad_enabled(True)
|
||||
|
||||
return matched_inputs, onnx_outputs
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user