code: def remove_unused_keys(state_dict):
      """remove unused keys (e.g.: seg_head.aux_head)"""
      keys_to_ignore = []
      for k in state_dict.keys():
          if k.startswith("seg_head.aux_head."):
              keys_to_ignore.append(k)
      for k in keys_to_ignore:
          state_dict.pop(k, None)
  
file_path: transformers/src/transformers/models/mobilevitv2/convert_mlcvnets_to_pytorch.py
imports: import argparse
  import collections
  import json
  from pathlib import Path
  import requests
  import torch
  import yaml
  from huggingface_hub import hf_hub_download
  from PIL import Image
  from transformers import MobileViTImageProcessor, MobileViTV2Config, MobileViTV2ForImageClassification, MobileViTV2ForSemanticSegmentation
  from transformers.utils import logging
label_desc: Use image node_id 5 for calling a global function with example usage: remove_unused_keys(state_dict) without return types