search

RitaRamo / smallcap

SmallCap: Lightweight Image Captioning Prompted with Retrieval Augmentation
88 stars 17 forks source link

VisionEncoderDecoderModel not found #1

Closed Twilighter9527 closed 1 year ago

Twilighter9527 commented 1 year ago

image there is no VisionEncoderDecoderModel in vision_encoder_decoder.py. when I read the file. I change it to "from transformers import VisionEncoderDecoderModel" in train.py . but there is another error. image thanks!

YovaKem commented 1 year ago

Should be resolved now (the risks of last minute refactoring.. sorry). Inference might still give you trouble - I'll test that out properly later this week.