Browse Source
update
Signed-off-by: junjie.jiang <junjie.jiang@zilliz.com>
main
1 changed files with
2 additions and
0 deletions
-
clip_vision.py
|
|
@ -17,9 +17,11 @@ import numpy as np |
|
|
|
from torchvision import transforms as T |
|
|
|
|
|
|
|
from towhee.operator import NNOperator, SharedType |
|
|
|
from towhee.dc2 import accelerate |
|
|
|
from towhee.models import clip |
|
|
|
|
|
|
|
|
|
|
|
@accelerate |
|
|
|
class Model: |
|
|
|
def __init__(self, model_name, device='cpu'): |
|
|
|
self.model = clip.create_model(model_name=model_name, pretrained=True, device=device).visual |
|
|
|