diff --git a/clip.py b/clip.py index a611d5f..f09bfc8 100644 --- a/clip.py +++ b/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