def test_training(self):
        """
        Tests that all VQA models can be trained on a single batch
        """
        for model_class in self.all_model_classes:
            model = model_class(self.model_tester.get_config())
            loss = model(**self.model_tester.prepare_config_and_inputs_for_common()[1], training=True).loss

            self.assertIsNotNone(loss, "Loss should not be None")

        Tests that all VQA models can be trained on a single batch
        
dpkd/transformers/tests/models/blip/test_modeling_tf_blip.py