I tried using the Huggingface wrapper and I hit a similar bug as described here: https://huggingface.co/OATML-Markslab/Tranception_Large/discussions/1