Browse Source
Remove from_config.
Signed-off-by: wxywb <xy.wang@zilliz.com>
main
wxywb
1 year ago
1 changed files with
1 additions and
2 deletions
-
clip.py
|
|
@ -41,8 +41,7 @@ def create_model(model_name, modality, checkpoint_path, device): |
|
|
if checkpoint_path is None: |
|
|
if checkpoint_path is None: |
|
|
hf_clip_model = CLIPModel.from_pretrained(model_name) |
|
|
hf_clip_model = CLIPModel.from_pretrained(model_name) |
|
|
else: |
|
|
else: |
|
|
hf_clip_config = CLIPModel.from_config(model_name) |
|
|
|
|
|
hf_clip_model = CLIPModel.from_pretrained(checkpoint_path, config=hf_clip_config) |
|
|
|
|
|
|
|
|
hf_clip_model = CLIPModel.from_pretrained(checkpoint_path) |
|
|
hf_clip_model.to(device) |
|
|
hf_clip_model.to(device) |
|
|
hf_clip_model.eval() |
|
|
hf_clip_model.eval() |
|
|
|
|
|
|
|
|
|
