src/transformers/models/gptsan_japanese/modeling_gptsan_japanese.py/GPTSanJapaneseBlock/__init__
class GPTSanJapaneseBlock:    def __init__(self, config, ext_layer=False):
        super().__init__()
        self.self_attn = GPTSanJapaneseLayerSelfAttention(config)
        self.feed_forward = GPTSanJapaneseLayerDenseFF(config) if ext_layer else GPTSanJapaneseLayerSparseFF(config)