 junjie.jiang
|
6ff206bd86
|
opset_version to 14
Signed-off-by: junjie.jiang <junjie.jiang@zilliz.com>
|
7 months ago |
|
junjie.jiang
|
527aff4f69
|
Check gpu is available
Signed-off-by: junjie.jiang <junjie.jiang@zilliz.com>
|
7 months ago |
 Jael Gu
|
c1927bd7cb
|
Avoid auto download given local checkpoint path
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
1 year ago |
|
Jael Gu
|
aea524a9d3
|
Fix input size assert issue
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
1 year ago |
|
Jael Gu
|
db6727933a
|
Allow to pass checkpoint path
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
1 year ago |
|
junjie.jiang
|
505b0b220d
|
enable accelerate
Signed-off-by: junjie.jiang <junjie.jiang@zilliz.com>
|
2 years ago |
ChengZi
|
dbf64995d0
|
enhance robustness of timm tfms
|
2 years ago |
|
Jael Gu
|
3a82fb4ebe
|
Fix for list inputs
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
junjie.jiang
|
86b69319fc
|
rm _types
Signed-off-by: junjie.jiang <junjie.jiang@zilliz.com>
|
2 years ago |
|
Jael Gu
|
af2086bdce
|
Update
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
f6033fcc81
|
Update device
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
158d86d00e
|
Fix onnx export device issue
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
b543b1d130
|
Set logging level
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
14e813281f
|
Add transform config
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
junjie.jiang
|
9dcd5723a6
|
delete Image dep
Signed-off-by: junjie.jiang <junjie.jiang@zilliz.com>
|
2 years ago |
|
Jael Gu
|
9ce147c9a9
|
Update model config with new method
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
21b72075da
|
Debug decorator
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
efb75f26b3
|
Add torch no_grad in inference
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
dae5eb0a4a
|
Fix config issue to support Triton
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
725ca7d383
|
Update
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
8c66702529
|
Update
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
03c85991e3
|
Update
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
c7881ca7a0
|
Fix post process
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
3f503747f3
|
Fix dim issue
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
0fcdf6d962
|
Fix op for vit models
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
8da18d800a
|
Fix issue for config
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
768f0068b8
|
Remove accelerate
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
c0bf53ea6e
|
Debug for tritonserve
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
c947c6ea84
|
Debug to support TritonServer
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
a8b3af6f10
|
Support TritonServe
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
f769e4a73d
|
Fix image mode issue
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
7ddc9b0f86
|
Add benchmark script
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
cf57a2215c
|
Fix onnx
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
c7e446550b
|
Update model list
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
fa0aac7037
|
Add onnx test
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
d815531df0
|
Add test for onnx
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
junjie.jiang
|
2989fee8f3
|
add schema info
Signed-off-by: junjie.jiang <junjie.jiang@zilliz.com>
|
2 years ago |
|
Jael Gu
|
254ea6162e
|
Update
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
7126190325
|
Update
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
d6516ba189
|
Allow list as input
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
|
Jael Gu
|
107529e868
|
Add save_model & suppported_model_names
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
gexy5
|
cb951f924b
|
add save_model
Signed-off-by: gexy5 <xinyu.ge@zilliz.com>
|
3 years ago |
|
gexy5
|
13d94c49b8
|
add save_model
Signed-off-by: gexy5 <xinyu.ge@zilliz.com>
|
3 years ago |
|
Jael Gu
|
f40a4fd06e
|
Align output name
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
3 years ago |
|
Jael Gu
|
e3e02e24ac
|
Optimize image type
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
3 years ago |
|
Jael Gu
|
bc098ab1aa
|
Update
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
3 years ago |
|
Jael Gu
|
7e51ef22d1
|
Update
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
3 years ago |
|
Jael Gu
|
3a1376aa30
|
Allow BGR
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
3 years ago |
|
Jael Gu
|
a04940fb30
|
Update
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
3 years ago |
|
Jael Gu
|
6a341403a7
|
Refactor
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
3 years ago |
