Instructions to use openai/imagegpt-medium with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use openai/imagegpt-medium with Transformers:
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("openai/imagegpt-medium", dtype="auto") - Notebooks
- Google Colab
- Kaggle
ImageGPT was probably not trained on ImageNet-21K as mentioned in the description.
#2
by pshishodia - opened
The paper doesn't explicitly mentions ImageNet-21K. Furthermore, the original code specifies 1K classes for ImageGPT https://github.com/openai/image-gpt/blob/c6af2ebf57e2460c71fefa53cd9054b060cf716d/src/run.py#L164C5-L166C1