diff --git a/transformers/modeling_utils.py b/transformers/modeling_utils.py index 309ba952..398172a8 100644 --- a/transformers/modeling_utils.py +++ b/transformers/modeling_utils.py @@ -740,7 +740,7 @@ class SequenceSummary(nn.Module): def __init__(self, config): super(SequenceSummary, self).__init__() - self.summary_type = config.summary_type if hasattr(config, 'summary_use_proj') else 'last' + self.summary_type = config.summary_type if hasattr(config, 'summary_type') else 'last' if self.summary_type == 'attn': # We should use a standard multi-head attention module with absolute positional embedding for that. # Cf. https://github.com/zihangdai/xlnet/blob/master/modeling.py#L253-L276