Граф коммитов

63 Коммитов

Автор SHA1 Сообщение Дата
Wenbing Li b708aeaaf4
add a disable_tokenizer option for the beam search process (#116)
* add an enable_tokenizer options

* a formal name
2021-06-29 15:05:14 -07:00
Wenbing Li 800e360ef3
add convert gpt-2 function in the bs script as well (#114) 2021-06-17 14:51:47 -07:00
Wenbing Li 352b2003bc
the full beam search post processing for GPT-2 model (#94)
* the full beam search post processing for GPT-2 model

* minor fixes
2021-05-24 23:05:34 -07:00
Wenbing Li 06c902253f
re-package onnx process code (#93)
* re-package onnx process code

* update the docs and examples as well.
2021-05-14 09:23:50 -07:00
Wenbing Li 3806e29421
rename the package to onnxruntime_extensions (#92)
* rename the package to onnxruntime_extensions

* fix the cmake file
2021-05-12 12:02:57 -07:00
Wenbing Li a563194d41
The README.md doc update for the release (#87)
* Update the README.md for the release.

* polish the code

* add the mobile section

* update the example code in the README.md

* correction on pytorch tutorial
2021-05-07 14:12:25 -07:00
Wenbing Li 5fa95c9485
The Pytorch custom op tutorial (#89) 2021-05-05 11:09:12 -07:00
Wenbing Li 2243847f22
The GPT-2 post processing code end-to-end example (#86)
* Support the tensor renaming for the embedded graph

* Add ORT verifying step in the conversion.

* make the gpt-e2e work

* Support the loop in mytorch

* gpt2 end-to-end works

* Polish the code and fix the unit test.
2021-04-28 10:56:30 -07:00
Wenbing Li d4b0f85203
Support 2-output GPT2 Tokenizer. (#77) 2021-03-15 11:12:26 -07:00
Wenbing Li aa846c3116
Support the pre/post processing Python and pytorch code snippets to be converted and be merged into the ONNX model. (#76)
* initial checkins

* restructure the implementation.

* refine the Python interface

* Finalize the interface.

* Add the custmop class for the customization.

* Test the eager_op with vector_to_string customop

* Refine the customop conversion interface.

* initial onnx builder

* Runable with incorrect result.

* reformat the onnx_ops calls

* a few of operators working on tracing

* handcraft all op conversion

* Add the unit testing for mytorch

* unit test passed.

* Add some documents...

* Move non-torch API into onnxruntime_customops.utils module.

* Fix the unit test issues.

* Fix some typos.
2021-03-12 10:39:21 -08:00
TomWildenhain-Microsoft 55e9c4965e
Renamed tf2onnx tutorial (#37)
* Renamed tf2onnx tutorial

* Added file for C++ tutorial
2021-01-08 10:58:29 -08:00
Wenbing Li 2dfd95e64b
rename the package name to onnxruntime-customops (#36)
* normalize the root package name in Python

* fixing build on Linux

* update the tutorial as well
2020-12-22 20:16:30 -08:00
TomWildenhain-Microsoft a7b4ff310d
Tom/add tutorial (#32)
* Added getting started instructions for Windows

Signed-off-by: Tom Wildenhain <tomwi@microsoft.com>

* Created a tutorial for converting models with custom ops. WIP

* Removed long outputs

* Changed to keras syntax and added setup instructions

Co-authored-by: Wenbing Li <10278425+wenbingl@users.noreply.github.com>
2020-12-10 16:35:28 -05:00