Browse Source
update the operator.
Signed-off-by: wxywb <xy.wang@zilliz.com>
main
wxywb
2 years ago
2 changed files with
4 additions and
1 deletions
-
README.md
-
blip.py
|
|
|
@ -39,7 +39,7 @@ text_pipe = ( |
|
|
|
.output('text', 'vec') |
|
|
|
) |
|
|
|
|
|
|
|
DataCollection(image_pipe('./teddy.jpg')).show() |
|
|
|
DataCollection(img_pipe('./teddy.jpg')).show() |
|
|
|
DataCollection(text_pipe('A teddybear on a skateboard in Times Square.')).show() |
|
|
|
|
|
|
|
``` |
|
|
|
|
|
|
|
@ -13,9 +13,12 @@ |
|
|
|
# limitations under the License. |
|
|
|
|
|
|
|
import sys |
|
|
|
import os |
|
|
|
from pathlib import Path |
|
|
|
|
|
|
|
import torch |
|
|
|
import logging |
|
|
|
import warnings |
|
|
|
from torch import nn |
|
|
|
|
|
|
|
from transformers import AutoProcessor, BlipForImageTextRetrieval |
|
|
|
|
