code: def __init__(self) -> None:
      self._model_config_yaml = "model_config.yaml"
      self._model_weights_ckpt = "model_weights.ckpt"
      self._model_extracted_dir = None
  
imports: from __future__ import annotations
  import os
  import shutil
  import tarfile
  import tempfile
  import uuid
  from typing import Optional, Set, Union
  import torch
  from omegaconf import DictConfig, OmegaConf
  from omegaconf.omegaconf import open_dict
  from pytorch_lightning.trainer.trainer import Trainer
  from nemo.core import classes
  from nemo.utils import logging, model_utils
  from nemo.utils.app_state import AppState
  from nemo.utils.get_rank import is_global_rank_zero
  from nemo.utils.model_utils import inject_model_parallel_rank
label_desc: Use image node_id 1 to create a new SaveRestoreConnector object  with example: obj = SaveRestoreConnector()