Method _encode

deepspeed/model_implementations/diffusers/vae.py:76–77

(self, x, return_dict=True)

        return self.static_encoder_output

    def _encode(self, x, return_dict=True):
        return self.vae.encode(x, return_dict=return_dict)

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