search

comfyanonymous / ComfyUI

The most powerful and modular diffusion model GUI, api and backend with a graph/nodes interface.
https://www.comfy.org/
GNU General Public License v3.0
52.31k stars 5.52k forks source link

Support Flux DoRA from SimpleTuner #4259

Open markuryy opened 1 month ago

markuryy commented 1 month ago

After training a DoRA for Flux.1 dev via SimpleTuner, loading the DoRA in ComfyUI using the LoraLoaderModelOnly node results in a number of "lora key not loaded" messages. I chose this node because the text encoders were not fine tuned I believe.

For testing I have uploaded the safetensors of the DoRA I am trying to load here

model weight dtype torch.float8_e4m3fn, manual cast: torch.bfloat16
model_type FLUX
lora key not loaded: transformer.single_transformer_blocks.0.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.0.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.0.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.1.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.1.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.1.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.10.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.10.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.10.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.11.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.11.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.11.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.12.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.12.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.12.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.13.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.13.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.13.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.14.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.14.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.14.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.15.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.15.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.15.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.16.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.16.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.16.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.17.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.17.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.17.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.18.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.18.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.18.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.19.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.19.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.19.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.2.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.2.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.2.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.20.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.20.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.20.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.21.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.21.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.21.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.22.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.22.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.22.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.23.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.23.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.23.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.24.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.24.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.24.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.25.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.25.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.25.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.26.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.26.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.26.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.27.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.27.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.27.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.28.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.28.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.28.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.29.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.29.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.29.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.3.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.3.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.3.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.30.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.30.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.30.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.31.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.31.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.31.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.32.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.32.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.32.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.33.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.33.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.33.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.34.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.34.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.34.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.35.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.35.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.35.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.36.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.36.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.36.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.37.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.37.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.37.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.4.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.4.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.4.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.5.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.5.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.5.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.6.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.6.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.6.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.7.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.7.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.7.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.8.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.8.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.8.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.9.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.9.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.single_transformer_blocks.9.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.0.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.0.attn.to_out.0.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.0.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.0.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.1.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.1.attn.to_out.0.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.1.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.1.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.10.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.10.attn.to_out.0.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.10.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.10.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.11.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.11.attn.to_out.0.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.11.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.11.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.12.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.12.attn.to_out.0.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.12.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.12.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.13.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.13.attn.to_out.0.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.13.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.13.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.14.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.14.attn.to_out.0.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.14.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.14.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.15.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.15.attn.to_out.0.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.15.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.15.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.16.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.16.attn.to_out.0.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.16.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.16.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.17.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.17.attn.to_out.0.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.17.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.17.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.18.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.18.attn.to_out.0.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.18.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.18.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.2.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.2.attn.to_out.0.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.2.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.2.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.3.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.3.attn.to_out.0.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.3.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.3.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.4.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.4.attn.to_out.0.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.4.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.4.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.5.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.5.attn.to_out.0.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.5.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.5.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.6.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.6.attn.to_out.0.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.6.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.6.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.7.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.7.attn.to_out.0.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.7.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.7.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.8.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.8.attn.to_out.0.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.8.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.8.attn.to_v.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.9.attn.to_k.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.9.attn.to_out.0.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.9.attn.to_q.lora_magnitude_vector
lora key not loaded: transformer.transformer_blocks.9.attn.to_v.lora_magnitude_vector
Model doesn't have a device attribute.
clip missing: ['text_projection.weight']
Requested to load FluxClipModel_
Loading 1 new model
Requested to load Flux
Loading 1 new model

Thank you for looking into this (again)

stepfunction83 commented 1 month ago

I can confirm that I am experiencing the same thing on my end.

bghira commented 1 month ago

@comfyanonymous i can confirm this is the key structure we will be sticking with and it is safe to support this one.

comfyanonymous commented 1 month ago

Can someone give me a dora of this format that is confirmed working with an example image?