code: def get_wrapped_compression_model(
      compression_model: CompressionModel, cfg: omegaconf.DictConfig
  ) -> CompressionModel:
      if hasattr(cfg, "interleave_stereo_codebooks"):
          if cfg.interleave_stereo_codebooks.use:
              kwargs = dict_from_config(cfg.interleave_stereo_codebooks)
              kwargs.pop("use")
              compression_model = InterleaveStereoCompressionModel(
                  compression_model, **kwargs
              )
      if hasattr(cfg, "compression_model_n_q"):
          if cfg.compression_model_n_q is not None:
              compression_model.set_num_codebooks(
                  cfg.compression_model_n_q
              )
      return compression_model
  
imports: import typing
  import audiocraft
  import omegaconf
  import torch
  from .encodec import CompressionModel, EncodecModel, InterleaveStereoCompressionModel
  from .lm import LMModel
  from .lm_magnet import MagnetLMModel
  from ..modules.codebooks_patterns import CodebooksPatternProvider, DelayedPatternProvider, MusicLMPattern, ParallelPatternProvider, UnrolledPatternProvider, CoarseFirstPattern
  from ..modules.conditioners import BaseConditioner, ChromaStemConditioner, CLAPEmbeddingConditioner, ConditionFuser, ConditioningProvider, LUTConditioner, T5Conditioner
  from .unet import DiffusionUnet
  from ..None import quantization
  from ..utils.utils import dict_from_config
  from ..modules.diffusion_schedule import MultiBandProcessor, SampleProcessor
label_desc: Use image node_id 12 for calling a global function with example usage: get_wrapped_compression_model(compression_model, cfg) and returns: compression_model