download.py

# 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)