Instructions to use google/vit-base-patch16-224-in21k with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use google/vit-base-patch16-224-in21k with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-feature-extraction", model="google/vit-base-patch16-224-in21k")# Load model directly from transformers import AutoImageProcessor, AutoModel processor = AutoImageProcessor.from_pretrained("google/vit-base-patch16-224-in21k") model = AutoModel.from_pretrained("google/vit-base-patch16-224-in21k") - Notebooks
- Google Colab
- Kaggle
Saving the model with jit script
#1
by i9like9music - opened
Is it possible to save this model with torch jit script? I tried and there was an error. Are there any ways to avoid it and save?
Hey @i9like9music , could you open an issue in the transformers repository with a reproducible code sample and the full stack-trace so that we may help?
Please also link the transformers issue here so that people facing the same error can find it.
Thank you!