code: def test_generated_tensor_dtype(
      self,
      test_name: str,
      model_name: str,
      framework: str,
      int_dtype: str,
      float_dtype: str,
  ):
      config = AutoConfig.from_pretrained(model_name)
      normalized_config_class = (
          NormalizedConfigManager.get_normalized_config_class(
              config.model_type
          )
      )
      normalized_config = normalized_config_class(config)
      input_generator = DummyVisionInputGenerator(
          task="image-classification",
          normalized_config=normalized_config,
          batch_size=2,
          num_channels=3,
          height=224,
          width=224,
      )
      self.validate_dtype_for_framework(
          input_generator, "pixel_values", framework, float_dtype
      )
      self.validate_dtype_for_framework(
          input_generator, "pixel_mask", framework, int_dtype
      )
  
file_path: optimum/tests/utils/test_dummpy_input_generators.py
imports: from contextlib import nullcontext
  from typing import TYPE_CHECKING, Tuple
  from unittest import TestCase
  import torch
  from parameterized import parameterized
  from transformers import AutoConfig
  from transformers.utils import is_tf_available
  from optimum.utils import DTYPE_MAPPER, DummyAudioInputGenerator, DummyTextInputGenerator, DummyVisionInputGenerator
  from optimum.utils.normalized_config import NormalizedConfigManager
  from optimum.utils.testing_utils import grid_parameters
label_desc: Use image node_id 4 for calling the GenerateDummy obj's underlying member method code with example usage: obj.test_generated_tensor_dtype(test_name, model_name, framework, int_dtype, float_dtype) without return types