1-Click INSTALL Ai-TOOLKIT WAN 2.2 LORA TRAINING!
Added 2025-08-31 18:17:22 +0000 UTCHey everyone! I've created a 1-click installer for the “ULTIMATE WAN 2.2 LORA TRAINING AI‑TOOLKIT WEBUI.” VIDEOS & IMAGES! Just fricking insane! This really is ONE of the best Ai model currently available and Ai-Toolkit makes training a Lora super ultra easy as well, basically the ultimate combination, perfect for complete beginners! With it you can train any LoRA you want for WAN: people, styles, and even full video LoRAs… it’s just nuts!
You can check out the video right here: https://youtu.be/bhBZHPvX7gI
The installer of course automates the entire install process!
LOCAL INSTALL!!! RECOMMENDED TO HAVE AT LEAST 24GB OF VRAM FOR THE WAN TEXT TO IMAGE LORA TRAINING! if it's an issue, definitely use the Runpod installer instead as recommended in the video!
Local install:
Requisites:
– CUDA 12.8 (RTX-50) OR CUDA 12.6 (RTX-40/30/20/GTX)
– Git, Python 3.10 64-bit, Node 18/20 LTS
1. Download the AI-TOOLKIT_AUTO_INSTALL.bat and place it in a folder WITHOUT any spaces in the path!!!
2. Run the bat file and select either option 1 or 2 depending on your GPU
3. Wait for the install to finish
4. ???
5. Profit 😎
Use then one of the launcher once you have closed the ui and you want to restart it.
RUNPOD:
1. Create an account if you haven't already: Runpod
2. Click on Pod (on the left side) then click deploy
3. Choose a 5090 GPU (if you intend on training a text to image lora OR a text to video lora with text encoder unloaded and 512 resolutions) otherwise choose the 6000 pro blackwell GPU, and chose the pytorch 2.8.0 then edit the template and choose 100GB for both the container and volume disk, then under expose ports, edit:
8888 into 8888, 8675. Then go into environment variables below, click add and then type:
AI_TOOLKIT_AUTH on the left then type a password on the right
then scroll then and deploy.
4. Go to my pods, wait for everything to finish and then click "connect", then "Connect to HTTP SERVICE port 8888"
5. Then drag and drop the AI-TOOLKIT_AUTO_INSTALL-RUNPOD.sh file on the left side of the UI then click on the "Terminal" icon on the right side on the UI
6. Copy and paste these two lines then press enter:
chmod +x AI-TOOLKIT_AUTO_INSTALL-RUNPOD.sh
./AI-TOOLKIT_AUTO_INSTALL-RUNPOD.sh
7. Wait for everything to be installed
8. Go back to the port 8675 and click on it to launch the webui then enter your password you created previously
IF YOU WANT TO RESTART THE UI AFTER STOPPING AND RESTARTING THE POD
Use the same
chmod +x AI-TOOLKIT_AUTO_INSTALL-RUNPOD.sh
./AI-TOOLKIT_AUTO_INSTALL-RUNPOD.sh
command as you did for the install
As always, supporting me on Patreon allows me to keep creating helpful resources like this for the community. Thank you for your support - now go have some fun😉!
Comments
Second this, please!
Shad Armstrong
2025-12-02 09:08:51 +0000 UTCHi. Could you please fix the auto installation for offline use ?
newtoai
2025-11-15 08:30:02 +0000 UTCHi, I get the following error running L40S on Runpod: Generating baseline samples before training Generating Images: 0%| | 0/10 [00:00 File "/workspace/ai-toolkit/run.py", line 120, in main()main() File "/workspace/ai-toolkit/run.py", line 108, in main File "/workspace/ai-toolkit/run.py", line 108, in main raise eraise e File "/workspace/ai-toolkit/run.py", line 96, in main File "/workspace/ai-toolkit/run.py", line 96, in main job.run()job.run() File "/workspace/ai-toolkit/jobs/ExtensionJob.py", line 22, in run File "/workspace/ai-toolkit/jobs/ExtensionJob.py", line 22, in run process.run()process.run() File "/workspace/ai-toolkit/jobs/process/BaseSDTrainProcess.py", line 2007, in run File "/workspace/ai-toolkit/jobs/process/BaseSDTrainProcess.py", line 2007, in run self.sample(self.step_num)self.sample(self.step_num) File "/workspace/ai-toolkit/extensions_built_in/sd_trainer/DiffusionTrainer.py", line 288, in sample File "/workspace/ai-toolkit/extensions_built_in/sd_trainer/DiffusionTrainer.py", line 288, in sample super().sample(step, is_first)super().sample(step, is_first) File "/workspace/ai-toolkit/jobs/process/BaseSDTrainProcess.py", line 367, in sample File "/workspace/ai-toolkit/jobs/process/BaseSDTrainProcess.py", line 367, in sample self.sd.generate_images(gen_img_config_list, sampler=sample_config.sampler)self.sd.generate_images(gen_img_config_list, sampler=sample_config.sampler) File "/workspace/ai-toolkit/venv/lib/python3.12/site-packages/torch/utils/_contextlib.py", line 116, in decorate_context File "/workspace/ai-toolkit/venv/lib/python3.12/site-packages/torch/utils/_contextlib.py", line 116, in decorate_context return func(*args, **kwargs)return func(*args, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/workspace/ai-toolkit/toolkit/models/base_model.py", line 645, in generate_images File "/workspace/ai-toolkit/toolkit/models/base_model.py", line 645, in generate_images img = self.generate_single_image(img = self.generate_single_image( ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/workspace/ai-toolkit/extensions_built_in/diffusion_models/wan22/wan22_14b_model.py", line 542, in generate_single_image File "/workspace/ai-toolkit/extensions_built_in/diffusion_models/wan22/wan22_14b_model.py", line 542, in generate_single_image output = pipeline(output = pipeline( ^^^^^^^^^^^^^^^^^^ File "/workspace/ai-toolkit/extensions_built_in/diffusion_models/wan22/wan22_pipeline.py", line 242, in __call__ File "/workspace/ai-toolkit/extensions_built_in/diffusion_models/wan22/wan22_pipeline.py", line 242, in __call__ noise_pred = current_model(noise_pred = current_model( ^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/workspace/ai-toolkit/venv/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1751, in _wrapped_call_impl File "/workspace/ai-toolkit/venv/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1751, in _wrapped_call_impl return self._call_impl(*args, **kwargs)return self._call_impl(*args, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/workspace/ai-toolkit/venv/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1762, in _call_impl File "/workspace/ai-toolkit/venv/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1762, in _call_impl return forward_call(*args, **kwargs)return forward_call(*args, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/workspace/ai-toolkit/venv/lib/python3.12/site-packages/diffusers/models/transformers/transformer_wan.py", line 663, in forward File "/workspace/ai-toolkit/venv/lib/python3.12/site-packages/diffusers/models/transformers/transformer_wan.py", line 663, in forward hidden_states = block(hidden_states, encoder_hidden_states, timestep_proj, rotary_emb)hidden_states = block(hidden_states, encoder_hidden_states, timestep_proj, rotary_emb) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/workspace/ai-toolkit/venv/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1751, in _wrapped_call_impl File "/workspace/ai-toolkit/venv/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1751, in _wrapped_call_impl return self._call_impl(*args, **kwargs)return self._call_impl(*args, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/workspace/ai-toolkit/venv/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1762, in _call_impl File "/workspace/ai-toolkit/venv/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1762, in _call_impl return forward_call(*args, **kwargs)return forward_call(*args, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/workspace/ai-toolkit/venv/lib/python3.12/site-packages/diffusers/models/transformers/transformer_wan.py", line 474, in forward File "/workspace/ai-toolkit/venv/lib/python3.12/site-packages/diffusers/models/transformers/transformer_wan.py", line 474, in forward self.scale_shift_table + temb.float()self.scale_shift_table + temb.float() ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~^^~~~~~~~~~~~~~~~~~~~~~~~~~~ RuntimeErrorRuntimeError: : Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!
Vaivij
2025-10-08 20:16:01 +0000 UTCtheres some issue when running this locally where it doesnt set up a venv the correct way and will fetch torchaudio etc. from global python, paste the error code into gemini/qwen etc. (not chatgpt doesnt work) and you will get a guide on getting it working
Nils Johan Anti
2025-10-07 19:29:03 +0000 UTCit is unreliable my 10 $ wasted for nothing you cant leave the pc or you wont know if its running or not ostris official is better n reliable
simon
2025-09-22 18:19:10 +0000 UTCI intend to run locally on my 4090. Appears to be a standalone install. Most of my python direct interaction is under comfuyi standalone, but I also have separate miniconda environments to support ST, etc. Do you suggest setting up a new conda environment before running your standalone bat installer (with your specified python version, and cuda 12.8 is already on my base Win 11 system). Anything I should know before jumping in and trying it?
James
2025-09-12 22:37:09 +0000 UTCCan u do it for Mimic too? Or a video how to install what u need for working. Because runpod is not my favourite. Quite expensive with time. Or anyone knows how i can install all and what i n eed to install to get wan workflow working on mimicpc comfy?
Senpai Ashua
2025-09-08 11:08:13 +0000 UTChello, i try to finish setup, but when i try to connect to htts 8675 port, i reiceve error: Bad gateway Error code 502, what can i do? Thanks
Florin-Mihai Cosma
2025-09-02 18:46:03 +0000 UTCAnyone else having issues running this throug the Run pod? I did exactly as the video showed, once i paste the 2 lines of code and hit Enter, it starts downloading everything then fails and just stops there??
JAMES M BOWEN
2025-09-02 18:11:21 +0000 UTC
