diff --git a/README.md b/README.md index c4eb860..a09526c 100644 --- a/README.md +++ b/README.md @@ -19,7 +19,6 @@ import torch import requests from PIL import Image from transformers import ViTFeatureExtractor, AutoTokenizer, VisionEncoderDecoderModel -from transformers.testing_utils import require_sentorch_device loc = "ydshieh/vit-gpt2-coco-en"