code: def test_albert():
      sub_registry = model_zoo.get_sub_registry("transformers_albert")
  
      for name, (
          model_fn,
          data_gen_fn,
          _,
          _,
          _,
      ) in sub_registry.items():
          model = model_fn()
          trace_model_and_compare_output(model, data_gen_fn)
  
imports: import pytest
  import torch
  from hf_tracer_utils import trace_model_and_compare_output
  from packaging import version
  from colossalai.testing import clear_cache_before_run
  from tests.kit.model_zoo import model_zoo
label_desc: Use image node_id 1 for calling a global function with example usage: test_albert() without return types