logo
Browse Source

update

Signed-off-by: junjie.jiang <junjie.jiang@zilliz.com>
main
junjie.jiang 2 years ago
parent
commit
e5e2873871
  1. 2
      clip_vision.py

2
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

Loading…
Cancel
Save