code: def store_base_vae(model):
      global base_vae, checkpoint_info
      if checkpoint_info != model.sd_checkpoint_info:
          assert not loaded_vae_file, "Trying to store non-base VAE!"
          base_vae = deepcopy(model.first_stage_model.state_dict())
          checkpoint_info = model.sd_checkpoint_info
  
imports: import os
  import collections
  from dataclasses import dataclass
  from modules import paths, shared, devices, script_callbacks, sd_models, extra_networks, lowvram, sd_hijack, hashes
  import glob
  from copy import deepcopy
label_desc: Use image node_id 4 for calling a global function with example usage: store_base_vae(model) without return types