ai-toolkit
ai-toolkit copied to clipboard
Z-image lora training error.
Running 1 job Error running job: cannot import name 'WanTransformer3DModel' from 'diffusers' (c:\ai\ai-toolkit\venv\Lib\site-packages\diffusers_init_.py) Error running on_error: cannot access local variable 'job' where it is not associated with a value
Result:
- 0 completed jobs
- 1 failure
========================================
Traceback (most recent call last):
Traceback (most recent call last):
File "c:\ai\ai-toolkit\run.py", line 120, in
File "c:\ai\ai-toolkit\run.py", line 120, in main()main() File "c:\ai\ai-toolkit\run.py", line 108, in main File "c:\ai\ai-toolkit\run.py", line 108, in main raise eraise e File "c:\ai\ai-toolkit\run.py", line 95, in main File "c:\ai\ai-toolkit\run.py", line 95, in main job = get_job(config_file, args.name)job = get_job(config_file, args.name) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "c:\ai\ai-toolkit\toolkit\job.py", line 28, in get_job File "c:\ai\ai-toolkit\toolkit\job.py", line 28, in get_job from jobs import ExtensionJobfrom jobs import ExtensionJob File "c:\ai\ai-toolkit\jobs_init_.py", line 1, in File "c:\ai\ai-toolkit\jobs_init_.py", line 1, in from .BaseJob import BaseJobfrom .BaseJob import BaseJob File "c:\ai\ai-toolkit\jobs\BaseJob.py", line 5, in File "c:\ai\ai-toolkit\jobs\BaseJob.py", line 5, in from jobs.process import BaseProcessfrom jobs.process import BaseProcess File "c:\ai\ai-toolkit\jobs\process_init_.py", line 8, in File "c:\ai\ai-toolkit\jobs\process_init_.py", line 8, in from .TrainSliderProcess import TrainSliderProcessfrom .TrainSliderProcess import TrainSliderProcess File "c:\ai\ai-toolkit\jobs\process\TrainSliderProcess.py", line 26, in File "c:\ai\ai-toolkit\jobs\process\TrainSliderProcess.py", line 26, in from .BaseSDTrainProcess import BaseSDTrainProcessfrom .BaseSDTrainProcess import BaseSDTrainProcess File "c:\ai\ai-toolkit\jobs\process\BaseSDTrainProcess.py", line 28, in File "c:\ai\ai-toolkit\jobs\process\BaseSDTrainProcess.py", line 28, in from toolkit.custom_adapter import CustomAdapterfrom toolkit.custom_adapter import CustomAdapter File "c:\ai\ai-toolkit\toolkit\custom_adapter.py", line 13, in File "c:\ai\ai-toolkit\toolkit\custom_adapter.py", line 13, in from toolkit.models.control_lora_adapter import ControlLoraAdapterfrom toolkit.models.control_lora_adapter import ControlLoraAdapter File "c:\ai\ai-toolkit\toolkit\models\control_lora_adapter.py", line 5, in File "c:\ai\ai-toolkit\toolkit\models\control_lora_adapter.py", line 5, in from toolkit.lora_special import LoRASpecialNetworkfrom toolkit.lora_special import LoRASpecialNetwork File "c:\ai\ai-toolkit\toolkit\lora_special.py", line 10, in File "c:\ai\ai-toolkit\toolkit\lora_special.py", line 10, in from diffusers import UNet2DConditionModel, PixArtTransformer2DModel, AuraFlowTransformer2DModel, WanTransformer3DModelfrom diffusers import UNet2DConditionModel, PixArtTransformer2DModel, AuraFlowTransformer2DModel, WanTransformer3DModel ImportErrorImportError: : cannot import name 'WanTransformer3DModel' from 'diffusers' (c:\ai\ai-toolkit\venv\Lib\site-packages\diffusers_init_.py)cannot import name 'WanTransformer3DModel' from 'diffusers' (c:\ai\ai-toolkit\venv\Lib\site-packages\diffusers_init_.py)
Open your command prompt (cmd or PowerShell).
- Activate the virtual environment: c:\ai\ai-toolkit\venv\Scripts\activate
- Upgrade diffusers: pip install -U diffusers
- Run your job again.
BUT after it not starting again, Blink with another cmd window and sleep in JOB STARTING. No errors on any terminal.
i encountered error with diffusers and find that we need diffusers 0.36.0.dev0. To get it, put requirements.txt file from this repo in python_embeded folder of Ai toolkit. Run venv, cd to python_embeded and run pip install -r requirements.txt. It will get new diffusers
another work around reinstall everything very simple from: https://github.com/Tavris1/AI-Toolkit-Easy-Install