Transformers How to use RationAI/vit-patch16-224-prostate with Transformers:
# pip install -U transformers accelerate
# Load model directly
from transformers import AutoModel
model = AutoModel.from_pretrained("RationAI/vit-patch16-224-prostate", device_map="auto") 
