[Refactor] unified interface for long ctx attn

xdit-project / xDiT

xDiT: A Scalable Inference Engine for Diffusion Transformers (DiTs) with Massive Parallelism

Apache License 2.0

726 stars 56 forks source link

Closed Eigensystem closed 2 weeks ago

feifeibear commented 3 weeks ago

@Eigensystem update the unitests in tests/core/test_xfuser_attn.py