from .vae2_2 import Wan2_2_VAE