How to use facebook/vit-msn-small with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-feature-extraction", model="facebook/vit-msn-small")
# Load model directly from transformers import AutoImageProcessor, AutoModel processor = AutoImageProcessor.from_pretrained("facebook/vit-msn-small") model = AutoModel.from_pretrained("facebook/vit-msn-small")
changed AutoFeatureExtractor to AutoImageProcessor, as ViTFeatureExtractor is going to be deprecated in version 5 of transformers
· Sign up or log in to comment