RobertaModel layer diagram

I’m making a diagram with the layers and operations in RobertaModel (see the picture below).
I wonder if I got the class right or if I’m missing something. I plan to make a post with the diagram when it’s done.