def __init__(self, *args, **kwargs):
    super(UnifiedLanguageModel, self).__init__(*args, **kwargs)
    self.with_mlm = (self.with_mlm or True)
def __init__(self, *args, **kwargs):
    super(UnifiedLanguageModel, self).__init__(*args, **kwargs)
    self.with_mlm = (self.with_mlm or True)