Method _decode

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

(self, x, return_dict=True)

        return self.static_decoder_output

    def _decode(self, x, return_dict=True):
        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