Hi, your paper is great. But I encountered problems when operating your code. Could you please provide the dependency environment of your code?

by the way, I met the following problem:

Traceback (most recent call last):
File "/home/chenhao/super-resolution/Inf-DiT/generate_t2i_sr.py", line 22, in
from dit.model import DiffusionEngine
File "/home/chenhao/super-resolution/Inf-DiT/dit/model.py", line 23, in
from flash_attn import flash_attn_qkvpacked_func, flash_attn_func
File "/home/chenhao/anaconda3/envs/Inf-DiT/lib/python3.10/site-packages/flash_attn/init.py", line 3, in
from flash_attn.flash_attn_interface import (
File "/home/chenhao/anaconda3/envs/Inf-DiT/lib/python3.10/site-packages/flash_attn/flash_attn_interface.py", line 10, in
import flash_attn_2_cuda as flash_attn_cuda
ImportError: libcudart.so.11.0: cannot open shared object file: No such file or directory
DONE on chenhao-b660-gaming-x-ddr4