def __init__(self, with_lm=True, **kwargs):
    super(T5_Decoder, self).__init__(**kwargs)
    self.with_lm = with_lm
def __init__(self, with_lm=True, **kwargs):
    super(T5_Decoder, self).__init__(**kwargs)
    self.with_lm = with_lm