Solved by AlekseyDorkin in post #2 
              
              
                I think that depends on the base model. So for example, a RoBERTa-based model is likely going to work just fine, while a ViT-based one won’t.