ImportError: cannot import name 'AuraFlowTransformer2DModel' from 'diffusers' and other errors. Just not usable.

[DarkAlchy]

Result:

• 0 completed jobs
• 1 failure ======================================== Traceback (most recent call last): File "D:\ai-toolkit\run.py", line 90, in main() File "D:\ai-toolkit\run.py", line 86, in main raise e File "D:\ai-toolkit\run.py", line 77, in main job = get_job(config_file, args.name) File "D:\ai-toolkit\toolkit\job.py", line 19, in get_job from jobs import TrainJob File "D:\ai-toolkit\jobs_init_.py", line 1, in from .BaseJob import BaseJob File "D:\ai-toolkit\jobs\BaseJob.py", line 5, in from jobs.process import BaseProcess File "D:\ai-toolkit\jobs\process_init_.py", line 8, in from .TrainSliderProcess import TrainSliderProcess File "D:\ai-toolkit\jobs\process\TrainSliderProcess.py", line 26, in from .BaseSDTrainProcess import BaseSDTrainProcess File "D:\ai-toolkit\jobs\process\BaseSDTrainProcess.py", line 29, in from toolkit.lora_special import LoRASpecialNetwork File "D:\ai-toolkit\toolkit\lora_special.py", line 10, in from diffusers import UNet2DConditionModel, PixArtTransformer2DModel, AuraFlowTransformer2DModel ImportError: cannot import name 'AuraFlowTransformer2DModel' from 'diffusers' (D:\ai-toolkit\venv\lib\site-packages\diffusers_init_.py)

[DarkAlchy]

This WIP does not work as I removed all those Pixart and aura flows and had to fight your includes to it finally starting to be hit with more errors.