| # import requests | |
| # from PIL import Image | |
| from transformers import AutoImageProcessor, AutoModel | |
| # url = "http://images.cocodataset.org/val2017/000000039769.jpg" | |
| # image = Image.open(requests.get(url, stream=True).raw) | |
| PATH = "." | |
| MODEL_NAME = "apple/aimv2-large-patch14-native" | |
| processor = AutoImageProcessor.from_pretrained(MODEL_NAME) | |
| processor.save_pretrained(PATH) | |
| model = AutoModel.from_pretrained( | |
| MODEL_NAME, | |
| trust_remote_code=True, | |
| ) | |
| model.save_pretrained(PATH) | |
| # inputs = processor(images=image, return_tensors="pt") | |
| # outputs = model(**inputs) | |