transformers

6efe3956 - Use channels first format for hidden states