From dc68f91c06a1ba6f15268e5b9c13ae7a7c514084 Mon Sep 17 00:00:00 2001 From: Ankur Singh Date: Mon, 27 Feb 2023 15:00:09 +0000 Subject: [PATCH] Updates README.md sample running code to remove FutureWarning deprecation error for `ViTFeatureExtractor`. (#14) - Updates README.md sample running code to remove FutureWarning deprecation error for `ViTFeatureExtractor`. (496b5b1656cab6f6f963d1db47724f3cc56e6c7a) Co-authored-by: E --- README.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index e6f9ad5..4f84a99 100644 --- a/README.md +++ b/README.md @@ -28,12 +28,12 @@ This is an image captioning model trained by @ydshieh in [flax ](https://github. ```python -from transformers import VisionEncoderDecoderModel, ViTFeatureExtractor, AutoTokenizer +from transformers import VisionEncoderDecoderModel, ViTImageProcessor, AutoTokenizer import torch from PIL import Image model = VisionEncoderDecoderModel.from_pretrained("nlpconnect/vit-gpt2-image-captioning") -feature_extractor = ViTFeatureExtractor.from_pretrained("nlpconnect/vit-gpt2-image-captioning") +feature_extractor = ViTImageProcessor.from_pretrained("nlpconnect/vit-gpt2-image-captioning") tokenizer = AutoTokenizer.from_pretrained("nlpconnect/vit-gpt2-image-captioning") device = torch.device("cuda" if torch.cuda.is_available() else "cpu")