src/transformers/models/big_bird/modeling_flax_big_bird.py/FlaxBigBirdForQuestionAnsweringModule/setup
class FlaxBigBirdForQuestionAnsweringModule:    def setup(self):
        self.config.num_labels = 2
        self.bert = FlaxBigBirdModule(self.config, dtype=self.dtype, add_pooling_layer=self.add_pooling_layer)
        self.qa_classifier = FlaxBigBirdForQuestionAnsweringHead(self.config, dtype=self.dtype)