code: def update_state_dict(state_dict):
      """Checks state_dict on load, updates state dict if needed."""
      state_dict = _update_transformers_to_freeze_module(state_dict)
      state_dict = _update_combiner_no_input_features(state_dict)
      state_dict = _update_combiner_no_device_tensor(state_dict)
      return state_dict
  
label_desc: Use image node_id 4 for calling a global function with example usage: update_state_dict(state_dict) and returns: state_dict