diff --git a/src/gluonnlp/models/bart.py b/src/gluonnlp/models/bart.py index fe61fab37e..6b011a7566 100644 --- a/src/gluonnlp/models/bart.py +++ b/src/gluonnlp/models/bart.py @@ -289,6 +289,7 @@ def from_cfg(cls, cfg, dtype=None, tie_weights=cfg.MODEL.tie_weights, data_norm=cfg.MODEL.data_norm, use_pooler=use_pooler, + classifier_activation=classifier_activation, attention_dropout=cfg.MODEL.attention_dropout, activation_dropout=cfg.MODEL.activation_dropout, dropout=cfg.MODEL.dropout,