Environment Setup Request

[wenqiwenqi1]

Could you please advise on how to set up the environment? Are there specific requirements for the versions of flash-attn, cache-dit, and torch?

[DefTruth]

please try the example in cache-dit: https://github.com/vipshop/cache-dit/blob/main/examples/pipeline/run_flux.py