针对DiT(Scalable Diffusion Models with Transformers)
问题:采用accelerate launch加速,提示“Module Not Found Error: No module named ‘timm’”。在环境中确实有timm包,import不报错。
原因:创建环境时采用conda create -n dit --clone old_env,导致原accelerate config指向旧环境,旧环境中无timm引起报错。
解决:
conda activate dit
pip uninstall accelerate
pip install accelerate
accelerate config