junjie.jiang
|
c0f26b5975
|
Check if cuda available
Signed-off-by: junjie.jiang <junjie.jiang@zilliz.com>
|
8 months ago |
wxywb
|
50b5003eab
|
Remove from_config.
Signed-off-by: wxywb <xy.wang@zilliz.com>
|
1 year ago |
wxywb
|
a704105378
|
Support loading trained weights path.
Signed-off-by: wxywb <xy.wang@zilliz.com>
|
1 year ago |
wxywb
|
76713536b1
|
make accelerate evaliable.
Signed-off-by: wxywb <xy.wang@zilliz.com>
|
2 years ago |
jinlingxu06
|
4e25e34aef
|
separate model arguments.
Signed-off-by: jinlingxu06 <jinling.xu@zilliz.com>
|
2 years ago |
jinlingxu06
|
8fd4d32e23
|
add logging for clip.
Signed-off-by: jinlingxu06 <jinling.xu@zilliz.com>
|
2 years ago |
junjie.jiang
|
be6635fae0
|
fix in triton
Signed-off-by: junjie.jiang <junjie.jiang@zilliz.com>
|
2 years ago |
wxywb
|
1b9025a4bb
|
use the corresponding device in the save method.
Signed-off-by: wxywb <xy.wang@zilliz.com>
|
2 years ago |
wxywb
|
efa84801fe
|
update the model with wrapper.
Signed-off-by: wxywb <xy.wang@zilliz.com>
|
2 years ago |
wxywb
|
e458a883c0
|
fix the wrapper of the model.
Signed-off-by: wxywb <xy.wang@zilliz.com>
|
2 years ago |
wxywb
|
3af58627d6
|
add default value for device.
Signed-off-by: wxywb <xy.wang@zilliz.com>
|
2 years ago |
wxywb
|
d0b811ca62
|
fix the crop_size for different version of huggingface.
Signed-off-by: wxywb <xy.wang@zilliz.com>
|
2 years ago |
wxywb
|
2b22bf59dd
|
update the readme.
Signed-off-by: wxywb <xy.wang@zilliz.com>
|
2 years ago |
wxywb
|
9ecdb950f8
|
:w
:update the clip for serialization.
Signed-off-by: wxywb <xy.wang@zilliz.com>
|
2 years ago |
wxywb
|
25c68be2ac
|
update the operator.
Signed-off-by: wxywb <xy.wang@zilliz.com>
|
2 years ago |
wxywb
|
69ee0f99b5
|
adjust for batch data input.
Signed-off-by: wxywb <xy.wang@zilliz.com>
|
2 years ago |
wxywb
|
ccd5499b68
|
init the op.
Signed-off-by: wxywb <xy.wang@zilliz.com>
|
3 years ago |