code: def get_model_cls(model_name: str) -> Type[base_model.BaseModel]:
      """Get the corresponding model class based on the model string.
  
      API:
      ```
          model_builder= get_model_cls('fully_connected')
          model = model_builder(config, ...)
      ```
  
      Args:
        model_name: str; Name of the model, e.g. 'fully_connected'.
  
      Returns:
        The model architecture (a flax Model) along with its default config.
      Raises:
        ValueError if model_name is unrecognized.
      """
      if model_name not in ALL_MODELS.keys():
          raise ValueError("Unrecognized model: {}".format(model_name))
      return ALL_MODELS[model_name]
  
file_path: scenic/scenic/model_lib/models.py
imports: from typing import Type
  from scenic.model_lib.base_models import base_model
  from scenic.projects.baselines import axial_resnet
  from scenic.projects.baselines import bit_resnet
  from scenic.projects.baselines import fully_connected
  from scenic.projects.baselines import hybrid_vit
  from scenic.projects.baselines import mixer
  from scenic.projects.baselines import resnet
  from scenic.projects.baselines import simple_cnn
  from scenic.projects.baselines import unet
  from scenic.projects.baselines import vit
label_desc: Use image node_id 1 for calling a global function with example usage: get_model_cls(model_name) and returns: ALL_MODELS