Add Z Image LoRA fine tuning support

[ParamThakkar123]

No description provided.

[codecov-commenter]

Codecov Report

✅ All modified and coverable lines are covered by tests.

📢 Thoughts on this report? Let us know!

[ParamThakkar123]

I am getting this error while running fine tuning task with any diffusion model on the diffusion trainer plugin:

terminate called after throwing an instance of 'std::length_error'
  what():  basic_string::_M_create

I tried fine tuning SDXL and other stable diffusion models but got this error on every run

[deep1401]

I am getting this error while running fine tuning task with any diffusion model on the diffusion trainer plugin:

terminate called after throwing an instance of 'std::length_error'
  what():  basic_string::_M_create

I tried fine tuning SDXL and other stable diffusion models but got this error on every run

I had this error once, updating timm resolved it. But it may or may not help in your case

[dadmobile]

I wasn't able to get this to running. Let's spend some time this week syncing on what's required and then we can do a patch release and announce this!

When trying to generate with Z Image Turbo I kept getting some vague error I will have to debug.

When trying to run the train I would get:

Error in Job: 'FlowMatchEulerDiscreteScheduler' object has no attribute 'add_noise'
Traceback (most recent call last):
  File "/home/azureuser/transformerlab-app/api/transformerlab/plugin_sdk/transformerlab/sdk/v1/tlab_plugin.py", line 105, in wrapper
    result = func(*args, **kwargs)
             ^^^^^^^^^^^^^^^^^^^^^
  File "/home/azureuser/.transformerlab/orgs/3c33c85b-628a-4ca8-93d3-b657cb7973b2/workspace/plugins/diffusion_trainer/main.py", line 818, in train_diffusion_lora
    noisy_latents = noise_scheduler.add_noise(latents, noise, timesteps)
                    ^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/azureuser/.transformerlab/orgs/3c33c85b-628a-4ca8-93d3-b657cb7973b2/workspace/plugins/diffusion_trainer/venv/lib/python3.11/site-packages/diffusers/configuration_utils.py", line 144, in __getattr__
    raise AttributeError(f"'{type(self).__name__}' object has no attribute '{name}'")
AttributeError: 'FlowMatchEulerDiscreteScheduler' object has no attribute 'add_noise'