logo

36 Commits (824ecf2154a4b666ee38e1819d4f0b1f50dcd162)

Author SHA1 Message Date
Jael Gu 824ecf2154 Remove pooler 2 years ago
Jael Gu 862459b19a Update forward method to support TritonServe 2 years ago
Jael Gu 082f59f277 Change accelerate to comment since not yet supported in latest release 2 years ago
ChengZi d2fb633df2 change train self.model to self._model 2 years ago
Jael Gu 7fa55ba6eb Debug for TritonServer 2 years ago
Jael Gu 5328585568 Modify save_model to support tritonserver 2 years ago
Jael Gu 9c26196077 Update op to support both models 2 years ago
Jael Gu e6c7f3d8e0 Support Towhee TritonServer 2 years ago
Jael Gu 0d30fc46a7 Add model 2 years ago
ChengZi ea58513a8f add relative package import 2 years ago
Jael Gu a00786d445 Update parameters 2 years ago
ChengZi f46ad5303f train mlm and clm task 2 years ago
Jael Gu a307a3196a Update model list 2 years ago
Jael Gu d838ba8357 Replace with transformers.onnx 2 years ago
Jael Gu 6d228847cf Update save_onnx 2 years ago
Jael Gu 9acb9072e1 Fix log 2 years ago
Jael Gu 1377e6a08e Fix issue due to enable_onnx_checker only available in torch 0.9 2 years ago
Jael Gu f0131944b4 Add log to test_onnx 2 years ago
Jael Gu 73fc63b013 Update for onnx test 2 years ago
Jael Gu 908e07a327 Add device 2 years ago
Jael Gu f5865f2021 Update model list for onnx 2 years ago
Jael Gu 976368836d Update todo 3 years ago
Jael Gu c6eabd756e Add list of supported_model_names(format='onnx') 3 years ago
Jael Gu a55e58b29b Update 3 years ago
Jael Gu c1351df718 Update 3 years ago
Jael Gu 516372f339 Allow to save model as onnx 3 years ago
Jael Gu 7082875b8b Debug 3 years ago
Jael Gu 2bc56f2cd6 Add supported_model_names 3 years ago
Jael Gu 71c4b9e93a Debug 3 years ago
Jael Gu 0bd1bcc127 Update model list 3 years ago
Jael Gu 0ee984c935 Modify save_model 3 years ago
Jael Gu ebfc9c74e3 Update 3 years ago
Jael Gu fd966f51bd Support save model with torchscript 3 years ago
Jael Gu 671ed01d78 Update 3 years ago
Jael Gu 1389d9923c Refactor 3 years ago
Jael Gu 923b3c6ff4 Refactor operator 3 years ago