code: def remove_module_from_state_dict(state_dict):
      new_state_dict = OrderedDict()
      for key, value in state_dict.items():
          name = "".join(key.split(".module"))
          new_state_dict[name] = value
      return new_state_dict
  
imports: import os
  import io
  import sys
  import json
  import torch
  import logging
  import numpy
  from collections import OrderedDict
  from pytorch_transformers.modeling_utils import CONFIG_NAME, WEIGHTS_NAME
  from tqdm import tqdm
  from blink.candidate_ranking.bert_reranking import BertReranker
  from blink.biencoder.biencoder import BiEncoderRanker
label_desc: Use image node_id 6 for calling a global function with example usage: remove_module_from_state_dict(state_dict) and returns: new_state_dict