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