search

xdit-project / xDiT

xDiT: A Scalable Inference Engine for Diffusion Transformers (DiTs) with Massive Parallelism
Apache License 2.0
720 stars 55 forks source link

fix: flux dp bug #353

Closed ictzyqq closed 3 days ago

ictzyqq commented 4 days ago

347 Fix TypeError when running Flux with DP and batchsize > 1.