Fix VAE spatial sharding dynamic calculation bug in Wan pipeline.#407

copybara-service[bot] merged 1 commit into

mainAI-Hypercomputer/maxdiffusion:mainfrom

ninatu/wan_vae_sharding_bugAI-Hypercomputer/maxdiffusion:ninatu/wan_vae_sharding_bugCopy head branch name to clipboard