Method _encode

wan/models/wan_vae.py:661–667

(self, x: torch.Tensor)

            raise ValueError("Invalid set gradient checkpointing")

    def _encode(self, x: torch.Tensor) -> torch.Tensor:
        x = [
            self.model.encode(u.unsqueeze(0), self.scale).squeeze(0)
            for u in x
        ]
        x = torch.stack(x)
        return x

    @apply_forward_hook
    def encode(