src/transformers/models/mbart/modeling_flax_mbart.py/FlaxMBartForSequenceClassificationModule/setup
class FlaxMBartForSequenceClassificationModule:    def setup(self):
        self.model = FlaxMBartModule(config=self.config, dtype=self.dtype)
        self.classification_head = FlaxMBartClassificationHead(
            config=self.config,
            inner_dim=self.config.d_model,
            num_classes=self.num_labels if self.num_labels is not None else self.config.num_labels,
            pooler_dropout=self.config.classifier_dropout,
        )