Method decode

wan/models/wan_vae.py:691–696

(self, z: torch.Tensor, return_dict: bool = True)

    @apply_forward_hook
    def decode(self, z: torch.Tensor, return_dict: bool = True) -> Union[DecoderOutput, torch.Tensor]:
        decoded = self._decode(z).sample

        if not return_dict:
            return (decoded,)
        return DecoderOutput(sample=decoded)

    @classmethod
    def from_pretrained(cls, pretrained_model_path, additional_kwargs={}):