NielsRogge / Transformers-Tutorials

This repository contains demos I made with the Transformers library by HuggingFace.
MIT License
8.63k stars 1.35k forks source link

'transformers.models.vision_encoder_decoder.configuration_vision_encoder_decoder.VisionEncoderDecoderConfig'> #179

Open arain60gb opened 1 year ago

arain60gb commented 1 year ago

kindly guide me how reslove this issue

loaded_model = VisionEncoderDecoderModel.from_pretrained('/content/drive/MyDrive/ocr_pth/checkpoint-5000') processor = TrOCRProcessor.from_pretrained("/content/drive/MyDrive/ocr_pth/checkpoint-5000")

KeyError Traceback (most recent call last) in 1 # loaded_preprocessor = TrOCRProcessor.from_pretrained('/content/drive/MyDrive/ocr_pth/checkpoint-5000') 2 loaded_model = VisionEncoderDecoderModel.from_pretrained('/content/drive/MyDrive/ocr_pth/checkpoint-5000') ----> 3 processor = TrOCRProcessor.from_pretrained("/content/drive/MyDrive/ocr_pth/checkpoint-5000")

3 frames /usr/local/lib/python3.7/dist-packages/transformers/models/auto/auto_factory.py in getitem(self, key) 570 model_name = self._model_mapping[mtype] 571 return self._load_attr_from_module(mtype, model_name) --> 572 raise KeyError(key) 573 574 def _load_attr_from_module(self, model_type, attr):

KeyError: <class 'transformers.models.vision_encoder_decoder.configuration_vision_encoder_decoder.VisionEncoderDecoderConfig'>

NithinJangid commented 1 year ago

Hi @zainali60 did you solve the problem? i am dealing with the same.