search

THUDM / Inf-DiT

Official implementation of Inf-DiT: Upsampling Any-Resolution Image with Memory-Efficient Diffusion Transformer
Apache License 2.0
378 stars 19 forks source link

dependency environment #24

Open hydrachen opened 3 months ago

hydrachen commented 3 months ago

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