 Jael Gu
|
2f48986bbe
|
Fix token count
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
10 months ago |
|
Jael Gu
|
1902ba932b
|
Output token usage
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
10 months ago |
 junjie.jiang
|
ba01b47853
|
Add transformers dep
Signed-off-by: junjie.jiang <junjie.jiang@zilliz.com>
|
1 year ago |
|
junjie.jiang
|
5eccfbc020
|
rm requirements.txt
Signed-off-by: junjie.jiang <junjie.jiang@zilliz.com>
|
1 year ago |
|
junjie.jiang
|
21e37c0b3e
|
Checkout device available
Signed-off-by: junjie.jiang <junjie.jiang@zilliz.com>
|
2 years ago |
ChengZi
|
884323d93d
|
export onnx using cpu model
Signed-off-by: ChengZi <chen.zhang@zilliz.com>
|
2 years ago |
|
ChengZi
|
a07c49e4ef
|
remove redundant to_device
Signed-off-by: ChengZi <chen.zhang@zilliz.com>
|
2 years ago |
|
ChengZi
|
37eadfa858
|
optimizer for supporting local checkpoint_path
Signed-off-by: ChengZi <chen.zhang@zilliz.com>
|
2 years ago |
|
ChengZi
|
862b980643
|
support local checkpoint_path
Signed-off-by: ChengZi <chen.zhang@zilliz.com>
|
2 years ago |
|
junjie.jiang
|
9d597ec7a5
|
op return float
Signed-off-by: junjie.jiang <junjie.jiang@zilliz.com>
|
2 years ago |
|
ChengZi
|
97f155fcaf
|
test onnx and qps
Signed-off-by: ChengZi <chen.zhang@zilliz.com>
|
2 years ago |
|
ChengZi
|
5feecc6f6d
|
refactor to support trt accelerate
Signed-off-by: ChengZi <chen.zhang@zilliz.com>
|
2 years ago |
|
ChengZi
|
87ea1727e1
|
refactor to support onnx
Signed-off-by: ChengZi <chen.zhang@zilliz.com>
|
2 years ago |
|
ChengZi
|
8aa80249c0
|
adapt when model output label num is 2
Signed-off-by: ChengZi <chen.zhang@zilliz.com>
|
2 years ago |
|
junjie.jiang
|
f4e0714d14
|
add device and sigmoid
Signed-off-by: junjie.jiang <junjie.jiang@zilliz.com>
|
2 years ago |
 shiyu22
|
4be43e5b9e
|
Update rerank
Signed-off-by: shiyu22 <shiyu.chen@zilliz.com>
|
2 years ago |
|
shiyu22
|
8d5520f5a7
|
Update rerank
Signed-off-by: shiyu22 <shiyu.chen@zilliz.com>
|
2 years ago |
|
shiyu22
|
ef89201e5b
|
Update README
Signed-off-by: shiyu22 <shiyu.chen@zilliz.com>
|
2 years ago |
|
shiyu22
|
8713010957
|
Add threshold
Signed-off-by: shiyu22 <shiyu.chen@zilliz.com>
|
2 years ago |
|
shiyu22
|
916f1f424d
|
Update output
Signed-off-by: shiyu22 <shiyu.chen@zilliz.com>
|
2 years ago |
|
shiyu22
|
e43a511c07
|
Update output
Signed-off-by: shiyu22 <shiyu.chen@zilliz.com>
|
2 years ago |
|
shiyu22
|
f5b2df224b
|
Add rerank op
Signed-off-by: shiyu22 <shiyu.chen@zilliz.com>
|
2 years ago |
 towhee
|
f7d3ab9f71
|
Initial commit
|
2 years ago |
