code: def match(self, module):
      return isinstance(module, self._orig_layer_class)
  
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 2 for calling the VAEPolicy obj's underlying member method code with example usage: obj.match(module) and returns: isinstance