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 |
Jael Gu
|
46d7875f6b
|
remove dc2
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
|
2 years ago |
jinlingxu06
|
4e25e34aef
|
separate model arguments.
Signed-off-by: jinlingxu06 <jinling.xu@zilliz.com>
|
2 years ago |
jinlingxu06
|
9799350ce2
|
add datasets in requirement.
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 |
jinlingxu06
|
5c7e5c574b
|
and last checkpoint check.
Signed-off-by: jinlingxu06 <jinling.xu@zilliz.com>
|
2 years ago |
wxywb
|
56d0fd9808
|
fix the readme.
Signed-off-by: wxywb <xy.wang@zilliz.com>
|
2 years ago |
wxywb
|
5c8443d6ef
|
update the opearator.
Signed-off-by: wxywb <xy.wang@zilliz.com>
|
2 years ago |
wxywb
|
8f521ea2f6
|
remove test in the clip training.
Signed-off-by: wxywb <xy.wang@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
|
f43eed58c5
|
update the readme.
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
|
a8832471ff
|
update several sota models.
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
|
64cf357035
|
fix the reference paper link.
Signed-off-by: wxywb <xy.wang@zilliz.com>
|
3 years ago |
ChengZi
|
28ad04ff40
|
add towhee requirements
|
3 years ago |
wxywb
|
ccd5499b68
|
init the op.
Signed-off-by: wxywb <xy.wang@zilliz.com>
|
3 years ago |
image-text-embedding
|
2857301334
|
Initial commit
|
3 years ago |