Load OneTrainer Flux LoRas and DoRas

[ejektaflex]

Feature Idea

It looks like the Flux branch for OneTrainer is nearing completion - and the results are looking great! However, it seems as though ComfyUI cannot load the resulting .safetensors file. Attached is a text file with all of the keys of the resulting DoRa.

no-work.txt

Existing Solutions

No response

Other

No response

[DarkViewAI]

+1 Same for me as well

[comfyanonymous]

Can I have an example file for this?

[ejektaflex]

I'll see what I can do - GitHub is limited to 25MB and my rank 1 DoRa was 26MB if I remember right 🥲

[comfyanonymous]

Just upload it on huggingface or civitai.

[ejektaflex]

Here you are:

https://github.com/ejektaflex/ot_dora_test

[ejektaflex]

I tried this after d043997 and I do indeed get working results with a LoRa, but not with a DoRa.

I updated the repo above with a non-fried example DoRa model.

Edit: Here is an image of what I get when I use a DoRa:

LoRas look fine, base model looks fine.

[comfyanonymous]

That's a problem with the dora file.

[Enyakk]

That's a problem with the dora file.

It seems to be a general problem. I've had the exact same pink noise effect opening it in ComfyUI with a DoRA trained out of Onetrainer.

[comfyanonymous]

should be fixed now.