Generated code snippets for Tranception_medium doesn't work
I've updated the Transformers library with the command pip install --upgrade transformers., and from its source with the command pip install git+https://github.com/huggingface/transformers.git but cant run the generated code snippets
pipe = pipeline("fill-mask", model="OATML-Markslab/Tranception_Medium")
or
Load model directly
from transformers import AutoModelWithLMHead
model = AutoModelWithLMHead.from_pretrained("OATML-Markslab/Tranception_Medium", dtype="auto")
KeyError Traceback (most recent call last)
/usr/local/lib/python3.12/dist-packages/transformers/models/auto/configuration_auto.py in from_pretrained(cls, pretrained_model_name_or_path, **kwargs)
1359 try:
-> 1360 config_class = CONFIG_MAPPING[config_dict["model_type"]]
1361 except KeyError:
3 frames
KeyError: 'tranception'
During handling of the above exception, another exception occurred:
ValueError Traceback (most recent call last)
/usr/local/lib/python3.12/dist-packages/transformers/models/auto/configuration_auto.py in from_pretrained(cls, pretrained_model_name_or_path, **kwargs)
1360 config_class = CONFIG_MAPPING[config_dict["model_type"]]
1361 except KeyError:
-> 1362 raise ValueError(
1363 f"The checkpoint you are trying to load has model type {config_dict['model_type']} "
1364 "but Transformers does not recognize this architecture. This could be because of an "
ValueError: The checkpoint you are trying to load has model type tranception but Transformers does not recognize this architecture. This could be because of an issue with the checkpoint, or because your version of Transformers is out of date.
You can update Transformers with the command pip install --upgrade transformers. If this does not work, and the checkpoint is very new, then there may not be a release version that supports this model yet. In this case, you can get the most up-to-date code by installing Transformers from source with the command pip install git+https://github.com/huggingface/transformers.git