UnboundLocalError: local variable 'network_module' referenced before assignment

[Chrisai26]

Hi I can run SDXL training in kohya but running flux I get this error:

Traceback (most recent call last): File "E:\kohya_ss2\kohya_ss\venv\lib\site-packages\gradio\queueing.py", line 536, in process_events response = await route_utils.call_process_api( File "E:\kohya_ss2\kohya_ss\venv\lib\site-packages\gradio\route_utils.py", line 321, in call_process_api output = await app.get_blocks().process_api( File "E:\kohya_ss2\kohya_ss\venv\lib\site-packages\gradio\blocks.py", line 1935, in process_api result = await self.call_function( File "E:\kohya_ss2\kohya_ss\venv\lib\site-packages\gradio\blocks.py", line 1520, in call_function prediction = await anyio.to_thread.run_sync( # type: ignore File "E:\kohya_ss2\kohya_ss\venv\lib\site-packages\anyio\to_thread.py", line 56, in run_sync return await get_async_backend().run_sync_in_worker_thread( File "E:\kohya_ss2\kohya_ss\venv\lib\site-packages\anyio\_backends\_asyncio.py", line 2177, in run_sync_in_worker_thread return await future File "E:\kohya_ss2\kohya_ss\venv\lib\site-packages\anyio\_backends\_asyncio.py", line 859, in run result = context.run(func, *args) File "E:\kohya_ss2\kohya_ss\venv\lib\site-packages\gradio\utils.py", line 826, in wrapper response = f(*args, **kwargs) File "E:\kohya_ss2\kohya_ss\kohya_gui\lora_gui.py", line 1171, in train_model "network_module": network_module, UnboundLocalError: local variable 'network_module' referenced before assignment

I suppose this has to do with the venv and tutorial on how to set it up?

[Clusters]

Was there ever a solution to this? Having the same issue.

[Clusters]

I got it fixed!
You have to use the working branch for Flux called "sd3-flux.1", because "master" branch does not yet support Flux.
Results are amazing, give it a try.