code: def test_timm_models():
      torch.backends.cudnn.deterministic = True
  
      sub_model_zoo = model_zoo.get_sub_registry("timm")
  
      for name, (
          model_fn,
          data_gen_fn,
          output_transform_fn,
          _,
          _,
          attribute,
      ) in sub_model_zoo.items():
          data = data_gen_fn()
          if attribute is not None and attribute.has_control_flow:
              meta_args = {k: v.to("meta") for k, v in data.items()}
          else:
              meta_args = None
  
          trace_and_compare(
              model_fn, data, output_transform_fn, meta_args
          )
  
file_path: ColossalAI/tests/test_fx/test_tracer/test_timm_model/test_timm_model.py
imports: import pytest
  import torch
  from packaging import version
  from colossalai._analyzer.fx import symbolic_trace
  from colossalai.testing import clear_cache_before_run
  from tests.kit.model_zoo import model_zoo
label_desc: Use image node_id 2 for calling a global function with example usage: test_timm_models() without return types