 Jael Gu
|
0485da7399
|
Modify output format
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
9a254c6c3d
|
Add pad token
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
ee8ec608d5
|
Fix for model without pad
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
0db0a18ad0
|
Fix for model without pad
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
6279bd4d80
|
Support sentence embedding
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
43a21ce0b0
|
Update Readme for save_model
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
93ed7f5e5e
|
Fix model name issue
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
824ecf2154
|
Remove pooler
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
862459b19a
|
Update forward method to support TritonServe
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
082f59f277
|
Change accelerate to comment since not yet supported in latest release
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
ChengZi
|
d2fb633df2
|
change train self.model to self._model
|
2 years ago |
|
Jael Gu
|
7fa55ba6eb
|
Debug for TritonServer
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
5328585568
|
Modify save_model to support tritonserver
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
9c26196077
|
Update op to support both models
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
e6c7f3d8e0
|
Support Towhee TritonServer
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
0d30fc46a7
|
Add model
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
ChengZi
|
ea58513a8f
|
add relative package import
|
2 years ago |
|
Jael Gu
|
a00786d445
|
Update parameters
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
ChengZi
|
f46ad5303f
|
train mlm and clm task
|
2 years ago |
|
Jael Gu
|
a307a3196a
|
Update model list
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
d838ba8357
|
Replace with transformers.onnx
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
6d228847cf
|
Update save_onnx
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
9acb9072e1
|
Fix log
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
1377e6a08e
|
Fix issue due to enable_onnx_checker only available in torch 0.9
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
f0131944b4
|
Add log to test_onnx
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
73fc63b013
|
Update for onnx test
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
908e07a327
|
Add device
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
f5865f2021
|
Update model list for onnx
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
976368836d
|
Update todo
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
3 years ago |
|
Jael Gu
|
c6eabd756e
|
Add list of supported_model_names(format='onnx')
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
3 years ago |
|
Jael Gu
|
a55e58b29b
|
Update
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
3 years ago |
|
Jael Gu
|
c1351df718
|
Update
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
3 years ago |
|
Jael Gu
|
516372f339
|
Allow to save model as onnx
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
3 years ago |
|
Jael Gu
|
7082875b8b
|
Debug
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
3 years ago |
|
Jael Gu
|
2bc56f2cd6
|
Add supported_model_names
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
3 years ago |
|
Jael Gu
|
71c4b9e93a
|
Debug
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
3 years ago |
|
Jael Gu
|
0bd1bcc127
|
Update model list
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
3 years ago |
|
Jael Gu
|
0ee984c935
|
Modify save_model
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
3 years ago |
|
Jael Gu
|
ebfc9c74e3
|
Update
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
3 years ago |
|
Jael Gu
|
fd966f51bd
|
Support save model with torchscript
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
3 years ago |
|
Jael Gu
|
671ed01d78
|
Update
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
3 years ago |
|
Jael Gu
|
1389d9923c
|
Refactor
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
3 years ago |
|
Jael Gu
|
923b3c6ff4
|
Refactor operator
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
3 years ago |
