code: def vit_small_patch32_224(pretrained=True, **kwargs):
      """ViT-Small (ViT-S/32)"""
      from avalanche.models.timm_vit import _create_vision_transformer
  
      model_kwargs = dict(
          patch_size=32, embed_dim=384, depth=12, num_heads=6, **kwargs
      )
      model = _create_vision_transformer(
          "vit_small_patch32_224", pretrained=pretrained, **model_kwargs
      )
      return model
  
file_path: avalanche/avalanche/models/vit.py
imports: from avalanche.models import SimpleMLP
label_desc: Use image node_id 4 for calling a global function with example usage: vit_small_patch32_224(pretrained) and returns: model