code: def apply(self, module, enable_cuda_graph=True):
      # TODO(cmikeh2): Enable cuda graph should be an inference configuration
      return DSVAE(module, enable_cuda_graph=enable_cuda_graph)
  
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 4 for calling the VAEPolicy obj's underlying member method code with example usage: obj.apply(module, enable_cuda_graph) and returns: DSVAE