Consider integrating Causal Forcing?

[zhuhz22]

Amazing work! I was wondering if you’d be open to integrating our Causal Forcing. It’s an upgraded version under the exact same setup as Self-Forcing, and it substantially improves visual quality, motion dynamics, and instruction following. Our project page: https://thu-ml.github.io/CausalForcing.github.io; code: https://github.com/thu-ml/Causal-Forcing. You only need to replace the Self Forcing initialization .pt file with Causal Forcing for your Stage 2.

Would you be interested? Thanks a lot!