| from transformers import AutoModel, CLIPImageProcessor, CLIPTokenizer | |
| import torch | |
| model_name_or_path = "BAAI/EVA-CLIP-8B" | |
| image_size = 224 | |
| processor = CLIPImageProcessor.from_pretrained("openai/clip-vit-large-patch14") | |
| model = AutoModel.from_pretrained( | |
| model_name_or_path, | |
| torch_dtype=torch.bfloat16, | |
| trust_remote_code=True).to('cuda').eval() | |
| tokenizer = CLIPTokenizer.from_pretrained(model_name_or_path) |