Browse Source
update the taiyi requirement.
Signed-off-by: wxywb <xy.wang@zilliz.com>
main
wxywb
2 years ago
2 changed files with
3 additions and
5 deletions
-
requirements.txt
-
taiyi.py
|
|
@ -0,0 +1,3 @@ |
|
|
|
|
|
torch |
|
|
|
|
|
transformers |
|
|
|
|
|
towhee |
|
|
@ -15,11 +15,6 @@ import torch |
|
|
from transformers import BertForSequenceClassification, BertConfig, BertTokenizer |
|
|
from transformers import BertForSequenceClassification, BertConfig, BertTokenizer |
|
|
from transformers import CLIPProcessor, CLIPModel |
|
|
from transformers import CLIPProcessor, CLIPModel |
|
|
|
|
|
|
|
|
import sys |
|
|
|
|
|
from pathlib import Path |
|
|
|
|
|
import torch |
|
|
|
|
|
from torchvision import transforms |
|
|
|
|
|
|
|
|
|
|
|
from towhee.types.image_utils import to_pil |
|
|
from towhee.types.image_utils import to_pil |
|
|
from towhee.operator.base import NNOperator, OperatorFlag |
|
|
from towhee.operator.base import NNOperator, OperatorFlag |
|
|
from towhee.types.arg import arg, to_image_color |
|
|
from towhee.types.arg import arg, to_image_color |
|
|
|
