code: def test_use_cache_forward_with_attn_mask(self):
      (
          config,
          inputs_dict,
      ) = self.model_tester.prepare_config_and_inputs()
      for model_class in self.all_model_classes:
          self.model_tester.check_use_cache_forward_with_attn_mask(
              model_class, config, inputs_dict
          )
  
imports: import unittest
  import numpy
  from transformers import BloomConfig, BloomTokenizerFast, is_flax_available
  from transformers.testing_utils import require_flax, slow
  from ...generation.test_flax_utils import FlaxGenerationTesterMixin
  from ...test_modeling_flax_common import FlaxModelTesterMixin, ids_tensor
label_desc: Use image node_id 3 for calling the FlaxBloomModelTest obj's underlying member method code with example usage: obj.test_use_cache_forward_with_attn_mask() without return types