5fa1a76
1
You only need to add a specific head to the pretrained model to manipulate the hidden states into your desired output!