optimum
b14379e9 - Fix PyTorch tied weights being duplicated in the exported ONNX models (#1326)

Commit

1 year ago

Fix PyTorch tied weights being duplicated in the exported ONNX models (#1326) * fix tied weights * overwrite external data as well * add transpose rather on MatMul node and not Gather * update on review * Update optimum/exporters/onnx/base.py Co-authored-by: Joshua Lochner <admin@xenova.com> --------- Co-authored-by: Joshua Lochner <admin@xenova.com>

References

#1326 - Fix PyTorch tied weights being duplicated in the exported ONNX models

Author

fxmarty

Parents

a39b1f56

Files6

optimum
- exporters/onnx
  - __main__.py
  - base.py
  - config.py
- onnx
  - __init__.py
  - graph_transformations.py
  - transformations_utils.py

optimum b14379e9 - Fix PyTorch tied weights being duplicated in the exported ONNX models (#1326)

optimum
b14379e9 - Fix PyTorch tied weights being duplicated in the exported ONNX models (#1326)