Update README.md
This commit is contained in:
parent
f6d2d95250
commit
973da61a93
|
@ -19,7 +19,6 @@ import torch
|
||||||
import requests
|
import requests
|
||||||
from PIL import Image
|
from PIL import Image
|
||||||
from transformers import ViTFeatureExtractor, AutoTokenizer, VisionEncoderDecoderModel
|
from transformers import ViTFeatureExtractor, AutoTokenizer, VisionEncoderDecoderModel
|
||||||
from transformers.testing_utils import require_sentorch_device
|
|
||||||
|
|
||||||
|
|
||||||
loc = "ydshieh/vit-gpt2-coco-en"
|
loc = "ydshieh/vit-gpt2-coco-en"
|
||||||
|
|
Loading…
Reference in New Issue