ai-toolkit icon indicating copy to clipboard operation
ai-toolkit copied to clipboard

Z-image lora training error.

Open Maelstrom2014 opened this issue 1 month ago • 3 comments

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)

Maelstrom2014 avatar Nov 28 '25 18:11 Maelstrom2014

Open your command prompt (cmd or PowerShell).

  1. Activate the virtual environment: c:\ai\ai-toolkit\venv\Scripts\activate
  2. Upgrade diffusers: pip install -U diffusers
  3. 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.

Maelstrom2014 avatar Nov 28 '25 18:11 Maelstrom2014

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

magicalfourlimbs avatar Nov 28 '25 23:11 magicalfourlimbs

another work around reinstall everything very simple from: https://github.com/Tavris1/AI-Toolkit-Easy-Install

Maelstrom2014 avatar Nov 29 '25 08:11 Maelstrom2014