nttdataspain
/

vit-gpt2-stablediffusion2-lora

vision-encoder-decoder

image-text-to-text

Model card Files Files and versions

D0k-tor commited on Jun 26, 2023

Commit

d2407b6

·

1 Parent(s): 440ba15

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -29,7 +29,7 @@ It produces optimal results in a reasonable time. Moreover, its implementation w
 from transformers import ViTFeatureExtractor, AutoTokenizer, VisionEncoderDecoderModel
 # Model
-model_id = "nttdataspain/vit-gpt2-coco-lora"
 model = VisionEncoderDecoderModel.from_pretrained(model_id)
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 feature_extractor = ViTFeatureExtractor.from_pretrained(model_id)

 from transformers import ViTFeatureExtractor, AutoTokenizer, VisionEncoderDecoderModel
 # Model
+model_id = "nttdataspain/vit-gpt2-stablediffusion2-lora"
 model = VisionEncoderDecoderModel.from_pretrained(model_id)
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 feature_extractor = ViTFeatureExtractor.from_pretrained(model_id)