D0k-tor commited on
Commit
d2407b6
1 Parent(s): 440ba15

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -29,7 +29,7 @@ It produces optimal results in a reasonable time. Moreover, its implementation w
29
  from transformers import ViTFeatureExtractor, AutoTokenizer, VisionEncoderDecoderModel
30
 
31
  # Model
32
- model_id = "nttdataspain/vit-gpt2-coco-lora"
33
  model = VisionEncoderDecoderModel.from_pretrained(model_id)
34
  tokenizer = AutoTokenizer.from_pretrained(model_id)
35
  feature_extractor = ViTFeatureExtractor.from_pretrained(model_id)
 
29
  from transformers import ViTFeatureExtractor, AutoTokenizer, VisionEncoderDecoderModel
30
 
31
  # Model
32
+ model_id = "nttdataspain/vit-gpt2-stablediffusion2-lora"
33
  model = VisionEncoderDecoderModel.from_pretrained(model_id)
34
  tokenizer = AutoTokenizer.from_pretrained(model_id)
35
  feature_extractor = ViTFeatureExtractor.from_pretrained(model_id)