logo

91 Commits (83e096c6dbdad3f2412ca133ff460ed8169480e2)
 

Author SHA1 Message Date
Jael Gu 83e096c6db Remove sentence post_proc 2 years ago
Jael Gu 5862443e70 Update readme 2 years ago
Jael Gu 3085df8baf Update readme 2 years ago
Jael Gu 46cef9e983 Update example result 2 years ago
Jael Gu 0485da7399 Modify output format 2 years ago
Jael Gu 0635dd4419 Update README 2 years ago
Jael Gu 9a254c6c3d Add pad token 2 years ago
Jael Gu ee8ec608d5 Fix for model without pad 2 years ago
Jael Gu 0db0a18ad0 Fix for model without pad 2 years ago
Jael Gu 6279bd4d80 Support sentence embedding 2 years ago
ChengZi 06062500cc add training requirements 2 years ago
ChengZi 730c0aaae3 add training readme 2 years ago
Jael Gu 1d34b7d14c Update Readme 2 years ago
Jael Gu 43a21ce0b0 Update Readme for save_model 2 years ago
Jael Gu 93ed7f5e5e Fix model name issue 2 years ago
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 009ddda7fc remove redundant script. 2 years ago
ChengZi d2fb633df2 change train self.model to self._model 2 years ago
Jael Gu 7fa55ba6eb Debug for TritonServer 2 years ago
ChengZi 45c55a398c add print args 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 34b3d3f0b8 Update run.py with ann_search.milvus_client 2 years ago
Jael Gu 88086ae2a8 Fix run.py for new op.model 2 years ago
Jael Gu e6c7f3d8e0 Support Towhee TritonServer 2 years ago
ChengZi 44fac86041 lazy import evaluate and datasets for avoiding potential error. 2 years ago
Jael Gu 0f9661823b Update benchmark/run.py 2 years ago
Jael Gu 4ea7a7b9ec Update benchmark/run.py 2 years ago
Jael Gu b883739db2 Update benchmark/run.py 2 years ago
Jael Gu 2a39a1697d Update save onnx in evaluation 2 years ago
Jael Gu 2cf6fabd69 Update evaluation method 2 years ago
Jael Gu 0592e151a0 Resolve import issue 2 years ago
ChengZi 9345e3bfd3 lazy import Trainer for avoiding potential error. 2 years ago
Jael Gu 4f343f9441 Update evaluate with onnx path 2 years ago
Jael Gu 36653e1a14 Update onnx test 2 years ago
Jael Gu 870c57ecbd Update README 2 years ago
Jael Gu 0d30fc46a7 Add model 2 years ago
ChengZi ea58513a8f add relative package import 2 years ago
Jael Gu 98d84e87f9 Add evaluation 2 years ago
Jael Gu a00786d445 Update parameters 2 years ago
ChengZi f46ad5303f train mlm and clm task 2 years ago
Jael Gu 5351a1cbd5 Update README 2 years ago
Jael Gu 00e0c3bd4f Update README 2 years ago
Jael Gu a307a3196a Update model list 2 years ago
Jael Gu 59f19c7f17 Add test onnx 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