Method _decode

deepspeed/model_implementations/diffusers/vae.py:34–35

(self, x, return_dict=True, generator=None)

        return self.static_decoder_output

    def _decode(self, x, return_dict=True, generator=None):
        return self.vae.decode(x, return_dict=return_dict)

    def _create_cuda_graph_decoder(self, *inputs, **kwargs):
        # warmup to create the workspace and cublas handle