SIGN IN SIGN UP

Optimize WAN Pipeline and VAE Inference

- Implement spatial/temporal parallelism for WAN VAE
- Replace resize with jnp.repeat for upsampling
- Resolve eager execution regression in VACE pipeline using vae_encode_pass
- Add comprehensive sharding validation and fallbacks in VAE with max_logging
- Fix multi-host addressable data logic and TeaCache tracking bugs
- Optimize pipeline formatting to reduce host overhead
- Restore profiler trace dumping functionality in generate_wan.py
T
Toshi Pahadia committed
ccad66068353bda1706a06a3e918a9311f6061e5
Parent: dddc939