Browse Source
use huggingface hub name
Signed-off-by: junjie.jiang <junjie.jiang@zilliz.com>
main
1 changed files with
9 additions and
1 deletions
-
clip.py
|
|
|
@ -164,6 +164,10 @@ class Clip(NNOperator): |
|
|
|
config['clip_vit_base_patch32'] = 'openai/clip-vit-base-patch32' |
|
|
|
config['clip_vit_large_patch14'] = 'openai/clip-vit-large-patch14' |
|
|
|
config['clip_vit_large_patch14_336'] ='openai/clip-vit-large-patch14-336' |
|
|
|
config['clip-vit-base-patch16'] = 'openai/clip-vit-base-patch16' |
|
|
|
config['clip-vit-base-patch32'] = 'openai/clip-vit-base-patch32' |
|
|
|
config['clip-vit-large-patch14'] = 'openai/clip-vit-large-patch14' |
|
|
|
config['clip-vit-large-patch14-336'] ='openai/clip-vit-large-patch14-336' |
|
|
|
return config |
|
|
|
|
|
|
|
@property |
|
|
|
@ -180,7 +184,11 @@ class Clip(NNOperator): |
|
|
|
'clip_vit_base_patch16', |
|
|
|
'clip_vit_base_patch32', |
|
|
|
'clip_vit_large_patch14', |
|
|
|
'clip_vit_large_patch14_336' |
|
|
|
'clip_vit_large_patch14_336', |
|
|
|
'clip-vit-base-patch16', |
|
|
|
'clip-vit-base-patch32', |
|
|
|
'clip-vit-large-patch14', |
|
|
|
'clip-vit-large-patch14_336' |
|
|
|
] |
|
|
|
if format == None: |
|
|
|
model_list = full_list |
|
|
|
|
