mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-16 21:00:14 +00:00
a1f9847b23
Description: Add the extra param to match gelu in PyTorch in the contrib symbolic function Motivation and Context Why is this change required? What problem does it solve? The symbolic function in /onnxruntime/python/tools/pytorch_export_contrib_ops.py is missing a recently added parameter approximate. We add this parameter and use the exporter defined gelu if approximate is "tanh".
13 lines
271 B
Text
13 lines
271 B
Text
--pre
|
|
-f https://download.pytorch.org/whl/torch_stable.html
|
|
# transformers requires sklearn
|
|
sklearn
|
|
numpy==1.21.0
|
|
transformers==v2.10.0
|
|
torch==1.10.0+cu113
|
|
torchvision==0.11.1+cu113
|
|
torchtext==0.11.0
|
|
tensorboard>=2.2.0,<2.5.0
|
|
h5py
|
|
setuptools>=41.4.0
|
|
parameterized>=0.8.1
|