sformers model. 
This specifies the model architecture. If not provided, a `gin_file` has to be provided.)