code: def __init__(self):
      super().__init__()
      try:
          import diffusers
  
          if hasattr(diffusers.models.vae, "AutoencoderKL"):
              self._orig_layer_class = (
                  diffusers.models.vae.AutoencoderKL
              )
          else:
              # Diffusers >= 0.12.0 changes location of AutoencoderKL
              self._orig_layer_class = (
                  diffusers.models.autoencoder_kl.AutoencoderKL
              )
      except ImportError:
          self._orig_layer_class = None
  
file_path: deepspeed/deepspeed/module_inject/containers/vae.py
imports: from ..policy import DSPolicy
  from ...model_implementations.diffusers.vae import DSVAE
label_desc: Use image node_id 1 to create a new VAEPolicy object from inherited base classes: DSPolicy with example: obj = VAEPolicy()