src/transformers/models/gptj/modeling_tf_gptj.py/TFGPTJModel/__init__
class TFGPTJModel:    def __init__(self, config, *inputs, **kwargs):
        super().__init__(config, *inputs, **kwargs)
        self.transformer = TFGPTJMainLayer(config, name="transformer")