No module named 'custom_nodes'

[Robson1970]

Hi, here is my ComfyUI log.

[MusubiTuner WanLoRATrainer] TRAINER: ERROR: Could not import training library or find necessary components: No module named 'custom_nodes'
[MusubiTuner WanLoRATrainer] TRAINER: Traceback (most recent call last):
[MusubiTuner WanLoRATrainer] TRAINER: File "I:\ComfyUI_windows_portable\ComfyUI\custom_nodes\ComfyUI_Wan2_1_lora_trainer\run_wan_training.py", line 30, in
[MusubiTuner WanLoRATrainer] TRAINER: ModuleNotFoundError: No module named 'custom_nodes'
[MusubiTuner WanLoRATrainer] TRAINER: RESULT:ERROR:ImportError:No module named 'custom_nodes'
[MusubiTuner WanLoRATrainer] TRAINER: Traceback (most recent call last):
[MusubiTuner WanLoRATrainer] TRAINER: File "C:\Users\1\AppData\Local\Programs\Python\Python310\lib\runpy.py", line 196, in _run_module_as_main
[MusubiTuner WanLoRATrainer] TRAINER: return _run_code(code, main_globals, None,
[MusubiTuner WanLoRATrainer] TRAINER: File "C:\Users\1\AppData\Local\Programs\Python\Python310\lib\runpy.py", line 86, in run_code
[MusubiTuner WanLoRATrainer] TRAINER: exec(code, run_globals)
[MusubiTuner WanLoRATrainer] TRAINER: File "C:\Users\1\AppData\Local\Programs\Python\Python310\Scripts\accelerate.EXE_main.py", line 7, in
[MusubiTuner WanLoRATrainer] TRAINER: sys.exit(main())
[MusubiTuner WanLoRATrainer] TRAINER: File "C:\Users\1\AppData\Local\Programs\Python\Python310\lib\site-packages\accelerate\commands\accelerate_cli.py", line 50, in main
[MusubiTuner WanLoRATrainer] TRAINER: args.func(args)
[MusubiTuner WanLoRATrainer] TRAINER: File "C:\Users\1\AppData\Local\Programs\Python\Python310\lib\site-packages\accelerate\commands\launch.py", line 1213, in launch_command
[MusubiTuner WanLoRATrainer] TRAINER: simple_launcher(args)
[MusubiTuner WanLoRATrainer] TRAINER: File "C:\Users\1\AppData\Local\Programs\Python\Python310\lib\site-packages\accelerate\commands\launch.py", line 795, in simple_launcher
[MusubiTuner WanLoRATrainer] TRAINER: raise subprocess.CalledProcessError(returncode=process.returncode, cmd=cmd)

[jaimitoes]

Hi, looking at the log, it seems that you are not cloning the repository into the custom node folder of ComfyUi. El dom, 1 jun 2025 a las 14:24, Robson1970 ***@***.***>) escribió:

…

*Robson1970* created an issue (jaimitoes/ComfyUI_Wan2_1_lora_trainer#1) <#1> Hi, here is my ComfyUI log. [MusubiTuner WanLoRATrainer] TRAINER: ERROR: Could not import training library or find necessary components: No module named 'custom_nodes' [MusubiTuner WanLoRATrainer] TRAINER: Traceback (most recent call last): [MusubiTuner WanLoRATrainer] TRAINER: File "I:\ComfyUI_windows_portable\ComfyUI\custom_nodes\ComfyUI_Wan2_1_lora_trainer\run_wan_training.py", line 30, in [MusubiTuner WanLoRATrainer] TRAINER: ModuleNotFoundError: No module named 'custom_nodes' [MusubiTuner WanLoRATrainer] TRAINER: RESULT:ERROR:ImportError:No module named 'custom_nodes' [MusubiTuner WanLoRATrainer] TRAINER: Traceback (most recent call last): [MusubiTuner WanLoRATrainer] TRAINER: File "C:\Users\1\AppData\Local\Programs\Python\Python310\lib\runpy.py", line 196, in _run_module_as_main [MusubiTuner WanLoRATrainer] TRAINER: return _run_code(code, main_globals, None, [MusubiTuner WanLoRATrainer] TRAINER: File "C:\Users\1\AppData\Local\Programs\Python\Python310\lib\runpy.py", line 86, in *run_code [MusubiTuner WanLoRATrainer] TRAINER: exec(code, run_globals) [MusubiTuner WanLoRATrainer] TRAINER: File "C:\Users\1\AppData\Local\Programs\Python\Python310\Scripts\accelerate.EXE_main*.py", line 7, in [MusubiTuner WanLoRATrainer] TRAINER: sys.exit(main()) [MusubiTuner WanLoRATrainer] TRAINER: File "C:\Users\1\AppData\Local\Programs\Python\Python310\lib\site-packages\accelerate\commands\accelerate_cli.py", line 50, in main [MusubiTuner WanLoRATrainer] TRAINER: args.func(args) [MusubiTuner WanLoRATrainer] TRAINER: File "C:\Users\1\AppData\Local\Programs\Python\Python310\lib\site-packages\accelerate\commands\launch.py", line 1213, in launch_command [MusubiTuner WanLoRATrainer] TRAINER: simple_launcher(args) [MusubiTuner WanLoRATrainer] TRAINER: File "C:\Users\1\AppData\Local\Programs\Python\Python310\lib\site-packages\accelerate\commands\launch.py", line 795, in simple_launcher [MusubiTuner WanLoRATrainer] TRAINER: raise subprocess.CalledProcessError(returncode=process.returncode, cmd=cmd) — Reply to this email directly, view it on GitHub <#1>, or unsubscribe <https://github.com/notifications/unsubscribe-auth/ABUT7YWP2CN4PMAN5WFO5WL3BLWI7AVCNFSM6AAAAAB6LBDPWGVHI2DSMVQWIX3LMV43ASLTON2WKOZTGEYDOMJXGQZDANY> . You are receiving this because you are subscribed to this thread.Message ID: ***@***.***>

[jaimitoes]

Okay, i'm looking is in your custom node folder. How you are running Comfyui? It seems is not finding the custom node folder for some relativity path reason. try to create a Bat as the example in the github Readme. El dom, 1 jun 2025 a las 22:19, SoloStudio Tech ***@***.***>) escribió:

…

Hi, looking at the log, it seems that you are not cloning the repository into the custom node folder of ComfyUi. El dom, 1 jun 2025 a las 14:24, Robson1970 ***@***.***>) escribió: > *Robson1970* created an issue (jaimitoes/ComfyUI_Wan2_1_lora_trainer#1) > <#1> > > Hi, here is my ComfyUI log. > > [MusubiTuner WanLoRATrainer] TRAINER: ERROR: Could not import training > library or find necessary components: No module named 'custom_nodes' > [MusubiTuner WanLoRATrainer] TRAINER: Traceback (most recent call last): > [MusubiTuner WanLoRATrainer] TRAINER: File > "I:\ComfyUI_windows_portable\ComfyUI\custom_nodes\ComfyUI_Wan2_1_lora_trainer\run_wan_training.py", > line 30, in > [MusubiTuner WanLoRATrainer] TRAINER: ModuleNotFoundError: No module > named 'custom_nodes' > [MusubiTuner WanLoRATrainer] TRAINER: RESULT:ERROR:ImportError:No module > named 'custom_nodes' > [MusubiTuner WanLoRATrainer] TRAINER: Traceback (most recent call last): > [MusubiTuner WanLoRATrainer] TRAINER: File > "C:\Users\1\AppData\Local\Programs\Python\Python310\lib\runpy.py", line > 196, in _run_module_as_main > [MusubiTuner WanLoRATrainer] TRAINER: return _run_code(code, > main_globals, None, > [MusubiTuner WanLoRATrainer] TRAINER: File > "C:\Users\1\AppData\Local\Programs\Python\Python310\lib\runpy.py", line 86, > in > > *run_code [MusubiTuner WanLoRATrainer] TRAINER: exec(code, run_globals) > [MusubiTuner WanLoRATrainer] TRAINER: File > "C:\Users\1\AppData\Local\Programs\Python\Python310\Scripts\accelerate.EXE_main*.py", > line 7, in > [MusubiTuner WanLoRATrainer] TRAINER: sys.exit(main()) > [MusubiTuner WanLoRATrainer] TRAINER: File > "C:\Users\1\AppData\Local\Programs\Python\Python310\lib\site-packages\accelerate\commands\accelerate_cli.py", > line 50, in main > [MusubiTuner WanLoRATrainer] TRAINER: args.func(args) > [MusubiTuner WanLoRATrainer] TRAINER: File > "C:\Users\1\AppData\Local\Programs\Python\Python310\lib\site-packages\accelerate\commands\launch.py", > line 1213, in launch_command > [MusubiTuner WanLoRATrainer] TRAINER: simple_launcher(args) > [MusubiTuner WanLoRATrainer] TRAINER: File > "C:\Users\1\AppData\Local\Programs\Python\Python310\lib\site-packages\accelerate\commands\launch.py", > line 795, in simple_launcher > [MusubiTuner WanLoRATrainer] TRAINER: raise > subprocess.CalledProcessError(returncode=process.returncode, cmd=cmd) > > — > Reply to this email directly, view it on GitHub > <#1>, or > unsubscribe > <https://github.com/notifications/unsubscribe-auth/ABUT7YWP2CN4PMAN5WFO5WL3BLWI7AVCNFSM6AAAAAB6LBDPWGVHI2DSMVQWIX3LMV43ASLTON2WKOZTGEYDOMJXGQZDANY> > . > You are receiving this because you are subscribed to this thread.Message > ID: ***@***.***> >

[jaimitoes]

@Robson1970 Also i'm looking you are using python 3.10. The last version of ComfyUI is using python 3.12, you need to update your windows portable package.

[signerzwb]

see this and try to find out if you need to set the pythonpath in your system

C:\AI\comfyui\venv>python -c "from custom_nodes.ComfyUI_Wan2_1_lora_trainer import wan_train_network; print('OK')"
Traceback (most recent call last):
File "", line 1, in
ModuleNotFoundError: No module named 'custom_nodes'

C:\AI\comfyui\venv>set PYTHONPATH=C:\AI\comfyui && python -c "from custom_nodes.ComfyUI_Wan2_1_lora_trainer import wan_train_network; print('OK')"
Trying to import sageattention
Successfully imported sageattention
MusubiTuner] WAN 2.1 LORA TRAINER.
OK

[monotected]

sorry but still not work for me (
so i also add wan_train_network to path , rebuild confui to 3.12 python

[MusubiTuner WanLoRATrainer] TRAINER: ipex flag is deprecated, will be removed in Accelerate v1.10. From 2.7.0, PyTorch has all needed optimizations for Intel CPU and XPU.
[MusubiTuner WanLoRATrainer] TRAINER: ERROR: Could not import training library or find necessary components: No module named 'custom_nodes'
[MusubiTuner WanLoRATrainer] TRAINER: Traceback (most recent call last):
[MusubiTuner WanLoRATrainer] TRAINER: File "F:\WANNNNN\ComfyUI\custom_nodes\ComfyUI_Wan2_1_lora_trainer\run_wan_training.py", line 30, in
[MusubiTuner WanLoRATrainer] TRAINER: ModuleNotFoundError: No module named 'custom_nodes'
[MusubiTuner WanLoRATrainer] TRAINER: RESULT:ERROR:ImportError:No module named 'custom_nodes'
[MusubiTuner WanLoRATrainer] TRAINER: Traceback (most recent call last):
[MusubiTuner WanLoRATrainer] TRAINER: File "", line 198, in _run_module_as_main
[MusubiTuner WanLoRATrainer] TRAINER: File "", line 88, in run_code
[MusubiTuner WanLoRATrainer] TRAINER: File "C:\Users\0_0\AppData\Local\Packages\PythonSoftwareFoundation.Python.3.12_qbz5n2kfra8p0\LocalCache\local-packages\Python312\Scripts\accelerate.EXE_main.py", line 7, in
[MusubiTuner WanLoRATrainer] TRAINER: File "C:\Users\0_0\AppData\Local\Packages\PythonSoftwareFoundation.Python.3.12_qbz5n2kfra8p0\LocalCache\local-packages\Python312\site-packages\accelerate\commands\accelerate_cli.py", line 50, in main
[MusubiTuner WanLoRATrainer] TRAINER: args.func(args)
[MusubiTuner WanLoRATrainer] TRAINER: File "C:\Users\0_0\AppData\Local\Packages\PythonSoftwareFoundation.Python.3.12_qbz5n2kfra8p0\LocalCache\local-packages\Python312\site-packages\accelerate\commands\launch.py", line 1199, in launch_command
[MusubiTuner WanLoRATrainer] TRAINER: simple_launcher(args)
[MusubiTuner WanLoRATrainer] TRAINER: File "C:\Users\0_0\AppData\Local\Packages\PythonSoftwareFoundation.Python.3.12_qbz5n2kfra8p0\LocalCache\local-packages\Python312\site-packages\accelerate\commands\launch.py", line 785, in simple_launcher
[MusubiTuner WanLoRATrainer] TRAINER: raise subprocess.CalledProcessError(returncode=process.returncode, cmd=cmd)
[MusubiTuner WanLoRATrainer] TRAINER: subprocess.CalledProcessError: Command '['C:\Users\0_0\AppData\Local\Microsoft\WindowsApps\PythonSoftwareFoundation.Python.3.12_qbz5n2kfra8p0\python.exe', 'F:\WANNNNN\ComfyUI\custom_nodes\ComfyUI_Wan2_1_lora_trainer\run_wan_training.py', '--dataset_config', 'F:\WANNNNN\ComfyUI\output\musubi_tuner_configs\musubi_dataset_config_20250718-161532.toml', '--xformers', '--max_train_steps', '2048', '--max_train_epochs', '15', '--max_data_loader_n_workers', '2', '--persistent_data_loader_workers', '--seed', '876', '--gradient_checkpointing', '--gradient_accumulation_steps', '1', '--mixed_precision', 'bf16', '--sample_every_n_steps', '50', '--sample_at_first', '--sample_every_n_epochs', '0', '--sample_prompts', 'F:\WANNNNN\ComfyUI\temp\musubi_tuner_prompts\prompts_20250718-1

[jaimitoes]

@monotected Have you created the "Bat file" as the Readme example?

[monotected]

Sorry was out of internet connection for few days (

Yepp , starts from (on new install comfui , new sage ): @echo off
REM Load Visual Studio Build Tools for the Wan subprocess environment (Optional to use max pow with the musubi compile settings and memory nodes)
call "C:\Program Files\Microsoft Visual Studio\2022\Community\VC\Auxiliary\Build\vcvarsall.bat" amd64
REM Start ComfyUI
.\python_embeded\python.exe -s ComfyUI\main.py --windows-standalone-build --use-sage-attention --fast fp16_accumulation
pause

now i stuck at
[MusubiTuner MusubiMemorySettings] Generating memory settings.
[MusubiTuner MusubiMemorySettings] Generated settings: {'blocks_to_swap': 38}
[MusubiTuner MusubiCompileSettings] Generating compile settings.
[MusubiTuner MusubiCompileSettings] Generated settings: {'dynamo_backend': 'EAGER'}
[MusubiTuner WanLoRATrainer] execute_train_lora called. Trigger: error_trigger
Prompt executed in 79.49 seconds

[jaimitoes]

@monotected It seems to be an issue conecting nodes. Check the new workflows in the "example_workflows", download it and open again. Also there is an update for the tested environment in the Readme.

[jaimitoes]

REM Load Visual Studio Build Tools for the Wan subprocess environment (Optional to use max pow with the musubi compile settings and memory nodes)
call "C:\Program Files\Microsoft Visual Studio\2022\Community\VC\Auxiliary\Build\vcvarsall.bat" amd64
REM Start ComfyUI
.\python_embeded\python.exe -s ComfyUI\main.py --windows-standalone-build

This, don't use sage attention.

[sev222]

same error. Linux Ubuntu 24.04 with VENV

[sev222]

if run in console:

export PYTHONPATH="/app:$PYTHONPATH" 
/app/venv/bin/python /app/custom_nodes/ComfyUI_Wan2_1_lora_trainer/run_wan_training.py .....

then work. please fix.

[HuntingSuccubus]

add this line to run_wan_training.py
sys.path.insert(0, os.getcwd())

[unjadded]

@shivshankar11 added the line of code there to the top after the import calls and before the first try wrap but didn't work

[MusubiTuner WanLoRATrainer] TRAINER: ModuleNotFoundError: No module named 'custom_nodes'
[MusubiTuner WanLoRATrainer] TRAINER: RESULT:ERROR:ImportError:No module named 'custom_nodes'