code: def remove_classification_head_(state_dict):
      ignore_keys = ["head.weight", "head.bias"]
      for k in ignore_keys:
          state_dict.pop(k, None)
  
file_path: transformers/src/transformers/models/vilt/convert_vilt_original_to_pytorch.py
imports: import argparse
  import json
  from pathlib import Path
  import requests
  import torch
  from huggingface_hub import hf_hub_download
  from PIL import Image
  from transformers import BertTokenizer, ViltConfig, ViltForImageAndTextRetrieval, ViltForImagesAndTextClassification, ViltForMaskedLM, ViltForQuestionAnswering, ViltImageProcessor, ViltProcessor
  from transformers.utils import logging
label_desc: Use image node_id 3 for calling a global function with example usage: remove_classification_head_(state_dict) without return types