ai-toolkit icon indicating copy to clipboard operation
ai-toolkit copied to clipboard

lora key not loaded: transformer.single_transformer_blocks.

Open wuyinde opened this issue 1 year ago • 7 comments

I successfully trained a LoRA model, and the output sample images generated during training appear normal. However, when I try to load the fine-tuned LoRA model in ComfyUI, I encounter an error that says the LoRA weights cannot be loaded. lora key not loaded: lora key not loaded: transformer.single_transformer_blocks.0.norm.linear.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.0.norm.linear.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.0.proj_mlp.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.0.proj_mlp.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.0.proj_out.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.0.proj_out.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.1.norm.linear.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.1.norm.linear.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.1.proj_mlp.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.1.proj_mlp.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.1.proj_out.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.1.proj_out.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.10.norm.linear.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.10.norm.linear.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.10.proj_mlp.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.10.proj_mlp.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.10.proj_out.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.10.proj_out.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.11.norm.linear.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.11.norm.linear.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.11.proj_mlp.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.11.proj_mlp.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.11.proj_out.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.11.proj_out.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.12.norm.linear.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.12.norm.linear.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.12.proj_mlp.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.12.proj_mlp.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.12.proj_out.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.12.proj_out.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.13.norm.linear.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.13.norm.linear.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.13.proj_mlp.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.13.proj_mlp.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.13.proj_out.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.13.proj_out.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.14.norm.linear.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.14.norm.linear.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.14.proj_mlp.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.14.proj_mlp.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.14.proj_out.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.14.proj_out.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.15.norm.linear.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.15.norm.linear.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.15.proj_mlp.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.15.proj_mlp.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.15.proj_out.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.15.proj_out.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.16.norm.linear.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.16.norm.linear.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.16.proj_mlp.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.16.proj_mlp.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.16.proj_out.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.16.proj_out.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.17.norm.linear.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.17.norm.linear.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.17.proj_mlp.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.17.proj_mlp.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.17.proj_out.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.17.proj_out.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.18.norm.linear.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.18.norm.linear.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.18.proj_mlp.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.18.proj_mlp.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.18.proj_out.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.18.proj_out.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.19.norm.linear.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.19.norm.linear.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.19.proj_mlp.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.19.proj_mlp.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.19.proj_out.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.19.proj_out.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.2.norm.linear.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.2.norm.linear.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.2.proj_mlp.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.2.proj_mlp.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.2.proj_out.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.2.proj_out.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.20.norm.linear.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.20.norm.linear.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.20.proj_mlp.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.20.proj_mlp.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.20.proj_out.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.20.proj_out.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.21.norm.linear.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.21.norm.linear.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.21.proj_mlp.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.21.proj_mlp.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.21.proj_out.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.21.proj_out.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.22.norm.linear.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.22.norm.linear.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.22.proj_mlp.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.22.proj_mlp.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.22.proj_out.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.22.proj_out.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.23.norm.linear.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.23.norm.linear.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.23.proj_mlp.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.23.proj_mlp.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.23.proj_out.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.23.proj_out.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.24.norm.linear.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.24.norm.linear.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.24.proj_mlp.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.24.proj_mlp.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.24.proj_out.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.24.proj_out.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.25.norm.linear.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.25.norm.linear.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.25.proj_mlp.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.25.proj_mlp.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.25.proj_out.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.25.proj_out.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.26.norm.linear.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.26.norm.linear.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.26.proj_mlp.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.26.proj_mlp.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.26.proj_out.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.26.proj_out.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.27.norm.linear.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.27.norm.linear.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.27.proj_mlp.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.27.proj_mlp.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.27.proj_out.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.27.proj_out.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.28.norm.linear.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.28.norm.linear.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.28.proj_mlp.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.28.proj_mlp.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.28.proj_out.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.28.proj_out.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.29.norm.linear.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.29.norm.linear.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.29.proj_mlp.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.29.proj_mlp.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.29.proj_out.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.29.proj_out.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.3.norm.linear.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.3.norm.linear.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.3.proj_mlp.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.3.proj_mlp.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.3.proj_out.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.3.proj_out.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.30.norm.linear.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.30.norm.linear.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.30.proj_mlp.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.30.proj_mlp.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.30.proj_out.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.30.proj_out.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.31.norm.linear.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.31.norm.linear.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.31.proj_mlp.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.31.proj_mlp.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.31.proj_out.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.31.proj_out.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.32.norm.linear.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.32.norm.linear.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.32.proj_mlp.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.32.proj_mlp.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.32.proj_out.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.32.proj_out.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.33.norm.linear.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.33.norm.linear.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.33.proj_mlp.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.33.proj_mlp.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.33.proj_out.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.33.proj_out.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.34.norm.linear.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.34.norm.linear.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.34.proj_mlp.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.34.proj_mlp.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.34.proj_out.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.34.proj_out.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.35.norm.linear.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.35.norm.linear.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.35.proj_mlp.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.35.proj_mlp.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.35.proj_out.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.35.proj_out.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.36.norm.linear.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.36.norm.linear.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.36.proj_mlp.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.36.proj_mlp.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.36.proj_out.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.36.proj_out.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.37.norm.linear.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.37.norm.linear.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.37.proj_mlp.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.37.proj_mlp.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.37.proj_out.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.37.proj_out.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.4.norm.linear.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.4.norm.linear.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.4.proj_mlp.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.4.proj_mlp.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.4.proj_out.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.4.proj_out.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.5.norm.linear.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.5.norm.linear.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.5.proj_mlp.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.5.proj_mlp.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.5.proj_out.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.5.proj_out.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.6.norm.linear.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.6.norm.linear.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.6.proj_mlp.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.6.proj_mlp.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.6.proj_out.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.6.proj_out.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.7.norm.linear.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.7.norm.linear.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.7.proj_mlp.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.7.proj_mlp.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.7.proj_out.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.7.proj_out.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.8.norm.linear.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.8.norm.linear.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.8.proj_mlp.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.8.proj_mlp.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.8.proj_out.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.8.proj_out.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.9.norm.linear.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.9.norm.linear.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.9.proj_mlp.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.9.proj_mlp.lora_B.weight lora key not loaded: transformer.single_transformer_blocks.9.proj_out.lora_A.weight lora key not loaded: transformer.single_transformer_blocks.9.proj_out.lora_B.weight lora key not loaded: transformer.transformer_blocks.0.attn.to_add_out.lora_A.weight lora key not loaded: transformer.transformer_blocks.0.attn.to_add_out.lora_B.weight lora key not loaded: transformer.transformer_blocks.0.ff.net.0.proj.lora_A.weight lora key not loaded: transformer.transformer_blocks.0.ff.net.0.proj.lora_B.weight lora key not loaded: transformer.transformer_blocks.0.ff.net.2.lora_A.weight lora key not loaded: transformer.transformer_blocks.0.ff.net.2.lora_B.weight lora key not loaded: transformer.transformer_blocks.0.ff_context.net.0.proj.lora_A.weight lora key not loaded: transformer.transformer_blocks.0.ff_context.net.0.proj.lora_B.weight lora key not loaded: transformer.transformer_blocks.0.ff_context.net.2.lora_A.weight lora key not loaded: transformer.transformer_blocks.0.ff_context.net.2.lora_B.weight lora key not loaded: transformer.transformer_blocks.0.norm1.linear.lora_A.weight lora key not loaded: transformer.transformer_blocks.0.norm1.linear.lora_B.weight lora key not loaded: transformer.transformer_blocks.0.norm1_context.linear.lora_A.weight lora key not loaded: transformer.transformer_blocks.0.norm1_context.linear.lora_B.weight lora key not loaded: transformer.transformer_blocks.1.attn.to_add_out.lora_A.weight lora key not loaded: transformer.transformer_blocks.1.attn.to_add_out.lora_B.weight lora key not loaded: transformer.transformer_blocks.1.ff.net.0.proj.lora_A.weight lora key not loaded: transformer.transformer_blocks.1.ff.net.0.proj.lora_B.weight lora key not loaded: transformer.transformer_blocks.1.ff.net.2.lora_A.weight lora key not loaded: transformer.transformer_blocks.1.ff.net.2.lora_B.weight lora key not loaded: transformer.transformer_blocks.1.ff_context.net.0.proj.lora_A.weight lora key not loaded: transformer.transformer_blocks.1.ff_context.net.0.proj.lora_B.weight lora key not loaded: transformer.transformer_blocks.1.ff_context.net.2.lora_A.weight lora key not loaded: transformer.transformer_blocks.1.ff_context.net.2.lora_B.weight lora key not loaded: transformer.transformer_blocks.1.norm1.linear.lora_A.weight lora key not loaded: transformer.transformer_blocks.1.norm1.linear.lora_B.weight lora key not loaded: transformer.transformer_blocks.1.norm1_context.linear.lora_A.weight lora key not loaded: transformer.transformer_blocks.1.norm1_context.linear.lora_B.weight lora key not loaded: transformer.transformer_blocks.10.attn.to_add_out.lora_A.weight lora key not loaded: transformer.transformer_blocks.10.attn.to_add_out.lora_B.weight lora key not loaded: transformer.transformer_blocks.10.ff.net.0.proj.lora_A.weight lora key not loaded: transformer.transformer_blocks.10.ff.net.0.proj.lora_B.weight lora key not loaded: transformer.transformer_blocks.10.ff.net.2.lora_A.weight lora key not loaded: transformer.transformer_blocks.10.ff.net.2.lora_B.weight lora key not loaded: transformer.transformer_blocks.10.ff_context.net.0.proj.lora_A.weight lora key not loaded: transformer.transformer_blocks.10.ff_context.net.0.proj.lora_B.weight lora key not loaded: transformer.transformer_blocks.10.ff_context.net.2.lora_A.weight lora key not loaded: transformer.transformer_blocks.10.ff_context.net.2.lora_B.weight lora key not loaded: transformer.transformer_blocks.10.norm1.linear.lora_A.weight lora key not loaded: transformer.transformer_blocks.10.norm1.linear.lora_B.weight lora key not loaded: transformer.transformer_blocks.10.norm1_context.linear.lora_A.weight lora key not loaded: transformer.transformer_blocks.10.norm1_context.linear.lora_B.weight lora key not loaded: transformer.transformer_blocks.11.attn.to_add_out.lora_A.weight lora key not loaded: transformer.transformer_blocks.11.attn.to_add_out.lora_B.weight lora key not loaded: transformer.transformer_blocks.11.ff.net.0.proj.lora_A.weight lora key not loaded: transformer.transformer_blocks.11.ff.net.0.proj.lora_B.weight lora key not loaded: transformer.transformer_blocks.11.ff.net.2.lora_A.weight lora key not loaded: transformer.transformer_blocks.11.ff.net.2.lora_B.weight lora key not loaded: transformer.transformer_blocks.11.ff_context.net.0.proj.lora_A.weight lora key not loaded: transformer.transformer_blocks.11.ff_context.net.0.proj.lora_B.weight lora key not loaded: transformer.transformer_blocks.11.ff_context.net.2.lora_A.weight lora key not loaded: transformer.transformer_blocks.11.ff_context.net.2.lora_B.weight lora key not loaded: transformer.transformer_blocks.11.norm1.linear.lora_A.weight lora key not loaded: transformer.transformer_blocks.11.norm1.linear.lora_B.weight lora key not loaded: transformer.transformer_blocks.11.norm1_context.linear.lora_A.weight lora key not loaded: transformer.transformer_blocks.11.norm1_context.linear.lora_B.weight lora key not loaded: transformer.transformer_blocks.12.attn.to_add_out.lora_A.weight lora key not loaded: transformer.transformer_blocks.12.attn.to_add_out.lora_B.weight lora key not loaded: transformer.transformer_blocks.12.ff.net.0.proj.lora_A.weight lora key not loaded: transformer.transformer_blocks.12.ff.net.0.proj.lora_B.weight lora key not loaded: transformer.transformer_blocks.12.ff.net.2.lora_A.weight lora key not loaded: transformer.transformer_blocks.12.ff.net.2.lora_B.weight lora key not loaded: transformer.transformer_blocks.12.ff_context.net.0.proj.lora_A.weight lora key not loaded: transformer.transformer_blocks.12.ff_context.net.0.proj.lora_B.weight lora key not loaded: transformer.transformer_blocks.12.ff_context.net.2.lora_A.weight lora key not loaded: transformer.transformer_blocks.12.ff_context.net.2.lora_B.weight lora key not loaded: transformer.transformer_blocks.12.norm1.linear.lora_A.weight lora key not loaded: transformer.transformer_blocks.12.norm1.linear.lora_B.weight lora key not loaded: transformer.transformer_blocks.12.norm1_context.linear.lora_A.weight lora key not loaded: transformer.transformer_blocks.12.norm1_context.linear.lora_B.weight lora key not loaded: transformer.transformer_blocks.13.attn.to_add_out.lora_A.weight lora key not loaded: transformer.transformer_blocks.13.attn.to_add_out.lora_B.weight lora key not loaded: transformer.transformer_blocks.13.ff.net.0.proj.lora_A.weight lora key not loaded: transformer.transformer_blocks.13.ff.net.0.proj.lora_B.weight lora key not loaded: transformer.transformer_blocks.13.ff.net.2.lora_A.weight lora key not loaded: transformer.transformer_blocks.13.ff.net.2.lora_B.weight lora key not loaded: transformer.transformer_blocks.13.ff_context.net.0.proj.lora_A.weight lora key not loaded: transformer.transformer_blocks.13.ff_context.net.0.proj.lora_B.weight lora key not loaded: transformer.transformer_blocks.13.ff_context.net.2.lora_A.weight lora key not loaded: transformer.transformer_blocks.13.ff_context.net.2.lora_B.weight lora key not loaded: transformer.transformer_blocks.13.norm1.linear.lora_A.weight lora key not loaded: transformer.transformer_blocks.13.norm1.linear.lora_B.weight lora key not loaded: transformer.transformer_blocks.13.norm1_context.linear.lora_A.weight lora key not loaded: transformer.transformer_blocks.13.norm1_context.linear.lora_B.weight lora key not loaded: transformer.transformer_blocks.14.attn.to_add_out.lora_A.weight lora key not loaded: transformer.transformer_blocks.14.attn.to_add_out.lora_B.weight lora key not loaded: transformer.transformer_blocks.14.ff.net.0.proj.lora_A.weight lora key not loaded: transformer.transformer_blocks.14.ff.net.0.proj.lora_B.weight lora key not loaded: transformer.transformer_blocks.14.ff.net.2.lora_A.weight lora key not loaded: transformer.transformer_blocks.14.ff.net.2.lora_B.weight lora key not loaded: transformer.transformer_blocks.14.ff_context.net.0.proj.lora_A.weight lora key not loaded: transformer.transformer_blocks.14.ff_context.net.0.proj.lora_B.weight lora key not loaded: transformer.transformer_blocks.14.ff_context.net.2.lora_A.weight lora key not loaded: transformer.transformer_blocks.14.ff_context.net.2.lora_B.weight lora key not loaded: transformer.transformer_blocks.14.norm1.linear.lora_A.weight lora key not loaded: transformer.transformer_blocks.14.norm1.linear.lora_B.weight lora key not loaded: transformer.transformer_blocks.14.norm1_context.linear.lora_A.weight lora key not loaded: transformer.transformer_blocks.14.norm1_context.linear.lora_B.weight lora key not loaded: transformer.transformer_blocks.15.attn.to_add_out.lora_A.weight lora key not loaded: transformer.transformer_blocks.15.attn.to_add_out.lora_B.weight lora key not loaded: transformer.transformer_blocks.15.ff.net.0.proj.lora_A.weight lora key not loaded: transformer.transformer_blocks.15.ff.net.0.proj.lora_B.weight lora key not loaded: transformer.transformer_blocks.15.ff.net.2.lora_A.weight lora key not loaded: transformer.transformer_blocks.15.ff.net.2.lora_B.weight lora key not loaded: transformer.transformer_blocks.15.ff_context.net.0.proj.lora_A.weight lora key not loaded: transformer.transformer_blocks.15.ff_context.net.0.proj.lora_B.weight lora key not loaded: transformer.transformer_blocks.15.ff_context.net.2.lora_A.weight lora key not loaded: transformer.transformer_blocks.15.ff_context.net.2.lora_B.weight lora key not loaded: transformer.transformer_blocks.15.norm1.linear.lora_A.weight lora key not loaded: transformer.transformer_blocks.15.norm1.linear.lora_B.weight lora key not loaded: transformer.transformer_blocks.15.norm1_context.linear.lora_A.weight lora key not loaded: transformer.transformer_blocks.15.norm1_context.linear.lora_B.weight lora key not loaded: transformer.transformer_blocks.16.attn.to_add_out.lora_A.weight lora key not loaded: transformer.transformer_blocks.16.attn.to_add_out.lora_B.weight lora key not loaded: transformer.transformer_blocks.16.ff.net.0.proj.lora_A.weight lora key not loaded: transformer.transformer_blocks.16.ff.net.0.proj.lora_B.weight lora key not loaded: transformer.transformer_blocks.16.ff.net.2.lora_A.weight lora key not loaded: transformer.transformer_blocks.16.ff.net.2.lora_B.weight lora key not loaded: transformer.transformer_blocks.16.ff_context.net.0.proj.lora_A.weight lora key not loaded: transformer.transformer_blocks.16.ff_context.net.0.proj.lora_B.weight lora key not loaded: transformer.transformer_blocks.16.ff_context.net.2.lora_A.weight lora key not loaded: transformer.transformer_blocks.16.ff_context.net.2.lora_B.weight lora key not loaded: transformer.transformer_blocks.16.norm1.linear.lora_A.weight lora key not loaded: transformer.transformer_blocks.16.norm1.linear.lora_B.weight lora key not loaded: transformer.transformer_blocks.16.norm1_context.linear.lora_A.weight lora key not loaded: transformer.transformer_blocks.16.norm1_context.linear.lora_B.weight lora key not loaded: transformer.transformer_blocks.17.attn.to_add_out.lora_A.weight lora key not loaded: transformer.transformer_blocks.17.attn.to_add_out.lora_B.weight lora key not loaded: transformer.transformer_blocks.17.ff.net.0.proj.lora_A.weight lora key not loaded: transformer.transformer_blocks.17.ff.net.0.proj.lora_B.weight lora key not loaded: transformer.transformer_blocks.17.ff.net.2.lora_A.weight lora key not loaded: transformer.transformer_blocks.17.ff.net.2.lora_B.weight lora key not loaded: transformer.transformer_blocks.17.ff_context.net.0.proj.lora_A.weight lora key not loaded: transformer.transformer_blocks.17.ff_context.net.0.proj.lora_B.weight lora key not loaded: transformer.transformer_blocks.17.ff_context.net.2.lora_A.weight lora key not loaded: transformer.transformer_blocks.17.ff_context.net.2.lora_B.weight lora key not loaded: transformer.transformer_blocks.17.norm1.linear.lora_A.weight lora key not loaded: transformer.transformer_blocks.17.norm1.linear.lora_B.weight lora key not loaded: transformer.transformer_blocks.17.norm1_context.linear.lora_A.weight lora key not loaded: transformer.transformer_blocks.17.norm1_context.linear.lora_B.weight lora key not loaded: transformer.transformer_blocks.18.attn.to_add_out.lora_A.weight lora key not loaded: transformer.transformer_blocks.18.attn.to_add_out.lora_B.weight lora key not loaded: transformer.transformer_blocks.18.ff.net.0.proj.lora_A.weight lora key not loaded: transformer.transformer_blocks.18.ff.net.0.proj.lora_B.weight lora key not loaded: transformer.transformer_blocks.18.ff.net.2.lora_A.weight lora key not loaded: transformer.transformer_blocks.18.ff.net.2.lora_B.weight lora key not loaded: transformer.transformer_blocks.18.ff_context.net.0.proj.lora_A.weight lora key not loaded: transformer.transformer_blocks.18.ff_context.net.0.proj.lora_B.weight lora key not loaded: transformer.transformer_blocks.18.ff_context.net.2.lora_A.weight lora key not loaded: transformer.transformer_blocks.18.ff_context.net.2.lora_B.weight lora key not loaded: transformer.transformer_blocks.18.norm1.linear.lora_A.weight lora key not loaded: transformer.transformer_blocks.18.norm1.linear.lora_B.weight lora key not loaded: transformer.transformer_blocks.18.norm1_context.linear.lora_A.weight lora key not loaded: transformer.transformer_blocks.18.norm1_context.linear.lora_B.weight lora key not loaded: transformer.transformer_blocks.2.attn.to_add_out.lora_A.weight lora key not loaded: transformer.transformer_blocks.2.attn.to_add_out.lora_B.weight lora key not loaded: transformer.transformer_blocks.2.ff.net.0.proj.lora_A.weight lora key not loaded: transformer.transformer_blocks.2.ff.net.0.proj.lora_B.weight lora key not loaded: transformer.transformer_blocks.2.ff.net.2.lora_A.weight lora key not loaded: transformer.transformer_blocks.2.ff.net.2.lora_B.weight lora key not loaded: transformer.transformer_blocks.2.ff_context.net.0.proj.lora_A.weight lora key not loaded: transformer.transformer_blocks.2.ff_context.net.0.proj.lora_B.weight lora key not loaded: transformer.transformer_blocks.2.ff_context.net.2.lora_A.weight lora key not loaded: transformer.transformer_blocks.2.ff_context.net.2.lora_B.weight lora key not loaded: transformer.transformer_blocks.2.norm1.linear.lora_A.weight lora key not loaded: transformer.transformer_blocks.2.norm1.linear.lora_B.weight lora key not loaded: transformer.transformer_blocks.2.norm1_context.linear.lora_A.weight lora key not loaded: transformer.transformer_blocks.2.norm1_context.linear.lora_B.weight lora key not loaded: transformer.transformer_blocks.3.attn.to_add_out.lora_A.weight lora key not loaded: transformer.transformer_blocks.3.attn.to_add_out.lora_B.weight lora key not loaded: transformer.transformer_blocks.3.ff.net.0.proj.lora_A.weight lora key not loaded: transformer.transformer_blocks.3.ff.net.0.proj.lora_B.weight lora key not loaded: transformer.transformer_blocks.3.ff.net.2.lora_A.weight lora key not loaded: transformer.transformer_blocks.3.ff.net.2.lora_B.weight lora key not loaded: transformer.transformer_blocks.3.ff_context.net.0.proj.lora_A.weight lora key not loaded: transformer.transformer_blocks.3.ff_context.net.0.proj.lora_B.weight lora key not loaded: transformer.transformer_blocks.3.ff_context.net.2.lora_A.weight lora key not loaded: transformer.transformer_blocks.3.ff_context.net.2.lora_B.weight lora key not loaded: transformer.transformer_blocks.3.norm1.linear.lora_A.weight lora key not loaded: transformer.transformer_blocks.3.norm1.linear.lora_B.weight lora key not loaded: transformer.transformer_blocks.3.norm1_context.linear.lora_A.weight lora key not loaded: transformer.transformer_blocks.3.norm1_context.linear.lora_B.weight lora key not loaded: transformer.transformer_blocks.4.attn.to_add_out.lora_A.weight lora key not loaded: transformer.transformer_blocks.4.attn.to_add_out.lora_B.weight lora key not loaded: transformer.transformer_blocks.4.ff.net.0.proj.lora_A.weight lora key not loaded: transformer.transformer_blocks.4.ff.net.0.proj.lora_B.weight lora key not loaded: transformer.transformer_blocks.4.ff.net.2.lora_A.weight lora key not loaded: transformer.transformer_blocks.4.ff.net.2.lora_B.weight lora key not loaded: transformer.transformer_blocks.4.ff_context.net.0.proj.lora_A.weight lora key not loaded: transformer.transformer_blocks.4.ff_context.net.0.proj.lora_B.weight lora key not loaded: transformer.transformer_blocks.4.ff_context.net.2.lora_A.weight lora key not loaded: transformer.transformer_blocks.4.ff_context.net.2.lora_B.weight lora key not loaded: transformer.transformer_blocks.4.norm1.linear.lora_A.weight lora key not loaded: transformer.transformer_blocks.4.norm1.linear.lora_B.weight lora key not loaded: transformer.transformer_blocks.4.norm1_context.linear.lora_A.weight lora key not loaded: transformer.transformer_blocks.4.norm1_context.linear.lora_B.weight lora key not loaded: transformer.transformer_blocks.5.attn.to_add_out.lora_A.weight lora key not loaded: transformer.transformer_blocks.5.attn.to_add_out.lora_B.weight lora key not loaded: transformer.transformer_blocks.5.ff.net.0.proj.lora_A.weight lora key not loaded: transformer.transformer_blocks.5.ff.net.0.proj.lora_B.weight lora key not loaded: transformer.transformer_blocks.5.ff.net.2.lora_A.weight lora key not loaded: transformer.transformer_blocks.5.ff.net.2.lora_B.weight lora key not loaded: transformer.transformer_blocks.5.ff_context.net.0.proj.lora_A.weight lora key not loaded: transformer.transformer_blocks.5.ff_context.net.0.proj.lora_B.weight lora key not loaded: transformer.transformer_blocks.5.ff_context.net.2.lora_A.weight lora key not loaded: transformer.transformer_blocks.5.ff_context.net.2.lora_B.weight lora key not loaded: transformer.transformer_blocks.5.norm1.linear.lora_A.weight lora key not loaded: transformer.transformer_blocks.5.norm1.linear.lora_B.weight lora key not loaded: transformer.transformer_blocks.5.norm1_context.linear.lora_A.weight lora key not loaded: transformer.transformer_blocks.5.norm1_context.linear.lora_B.weight lora key not loaded: transformer.transformer_blocks.6.attn.to_add_out.lora_A.weight lora key not loaded: transformer.transformer_blocks.6.attn.to_add_out.lora_B.weight lora key not loaded: transformer.transformer_blocks.6.ff.net.0.proj.lora_A.weight lora key not loaded: transformer.transformer_blocks.6.ff.net.0.proj.lora_B.weight lora key not loaded: transformer.transformer_blocks.6.ff.net.2.lora_A.weight lora key not loaded: transformer.transformer_blocks.6.ff.net.2.lora_B.weight lora key not loaded: transformer.transformer_blocks.6.ff_context.net.0.proj.lora_A.weight lora key not loaded: transformer.transformer_blocks.6.ff_context.net.0.proj.lora_B.weight lora key not loaded: transformer.transformer_blocks.6.ff_context.net.2.lora_A.weight lora key not loaded: transformer.transformer_blocks.6.ff_context.net.2.lora_B.weight lora key not loaded: transformer.transformer_blocks.6.norm1.linear.lora_A.weight lora key not loaded: transformer.transformer_blocks.6.norm1.linear.lora_B.weight lora key not loaded: transformer.transformer_blocks.6.norm1_context.linear.lora_A.weight lora key not loaded: transformer.transformer_blocks.6.norm1_context.linear.lora_B.weight lora key not loaded: transformer.transformer_blocks.7.attn.to_add_out.lora_A.weight lora key not loaded: transformer.transformer_blocks.7.attn.to_add_out.lora_B.weight lora key not loaded: transformer.transformer_blocks.7.ff.net.0.proj.lora_A.weight lora key not loaded: transformer.transformer_blocks.7.ff.net.0.proj.lora_B.weight lora key not loaded: transformer.transformer_blocks.7.ff.net.2.lora_A.weight lora key not loaded: transformer.transformer_blocks.7.ff.net.2.lora_B.weight lora key not loaded: transformer.transformer_blocks.7.ff_context.net.0.proj.lora_A.weight lora key not loaded: transformer.transformer_blocks.7.ff_context.net.0.proj.lora_B.weight lora key not loaded: transformer.transformer_blocks.7.ff_context.net.2.lora_A.weight lora key not loaded: transformer.transformer_blocks.7.ff_context.net.2.lora_B.weight lora key not loaded: transformer.transformer_blocks.7.norm1.linear.lora_A.weight lora key not loaded: transformer.transformer_blocks.7.norm1.linear.lora_B.weight lora key not loaded: transformer.transformer_blocks.7.norm1_context.linear.lora_A.weight lora key not loaded: transformer.transformer_blocks.7.norm1_context.linear.lora_B.weight lora key not loaded: transformer.transformer_blocks.8.attn.to_add_out.lora_A.weight lora key not loaded: transformer.transformer_blocks.8.attn.to_add_out.lora_B.weight lora key not loaded: transformer.transformer_blocks.8.ff.net.0.proj.lora_A.weight lora key not loaded: transformer.transformer_blocks.8.ff.net.0.proj.lora_B.weight lora key not loaded: transformer.transformer_blocks.8.ff.net.2.lora_A.weight lora key not loaded: transformer.transformer_blocks.8.ff.net.2.lora_B.weight lora key not loaded: transformer.transformer_blocks.8.ff_context.net.0.proj.lora_A.weight lora key not loaded: transformer.transformer_blocks.8.ff_context.net.0.proj.lora_B.weight lora key not loaded: transformer.transformer_blocks.8.ff_context.net.2.lora_A.weight lora key not loaded: transformer.transformer_blocks.8.ff_context.net.2.lora_B.weight lora key not loaded: transformer.transformer_blocks.8.norm1.linear.lora_A.weight lora key not loaded: transformer.transformer_blocks.8.norm1.linear.lora_B.weight lora key not loaded: transformer.transformer_blocks.8.norm1_context.linear.lora_A.weight lora key not loaded: transformer.transformer_blocks.8.norm1_context.linear.lora_B.weight lora key not loaded: transformer.transformer_blocks.9.attn.to_add_out.lora_A.weight lora key not loaded: transformer.transformer_blocks.9.attn.to_add_out.lora_B.weight lora key not loaded: transformer.transformer_blocks.9.ff.net.0.proj.lora_A.weight lora key not loaded: transformer.transformer_blocks.9.ff.net.0.proj.lora_B.weight lora key not loaded: transformer.transformer_blocks.9.ff.net.2.lora_A.weight lora key not loaded: transformer.transformer_blocks.9.ff.net.2.lora_B.weight lora key not loaded: transformer.transformer_blocks.9.ff_context.net.0.proj.lora_A.weight lora key not loaded: transformer.transformer_blocks.9.ff_context.net.0.proj.lora_B.weight lora key not loaded: transformer.transformer_blocks.9.ff_context.net.2.lora_A.weight lora key not loaded: transformer.transformer_blocks.9.ff_context.net.2.lora_B.weight lora key not loaded: transformer.transformer_blocks.9.norm1.linear.lora_A.weight lora key not loaded: transformer.transformer_blocks.9.norm1.linear.lora_B.weight lora key not loaded: transformer.transformer_blocks.9.norm1_context.linear.lora_A.weight lora key not loaded: transformer.transformer_blocks.9.norm1_context.linear.lora_B.weight Here are the key parameters I used during training:

  • type: 'sd_trainer' device: cuda:0 network: type: "lora" linear: 16 linear_alpha: 16 save: dtype: float16 train: dtype: bf16 model: name_or_path: "black-forest-labs/FLUX.1-dev"

wuyinde avatar Aug 15 '24 14:08 wuyinde

Having Same issues!

pranay0318 avatar Aug 15 '24 20:08 pranay0318

Update to the Latest Version of comfyui If you are on branch, git checkout master

pranay0318 avatar Aug 16 '24 05:08 pranay0318

use this script to make your LoRA compatible with comfy (and everything else): https://huggingface.co/comfyanonymous/flux_RealismLora_converted_comfyui/blob/main/convert.py

martintomov avatar Aug 16 '24 12:08 martintomov

@martintomov still experiencing after updating comfyui to the latest version and converting the lora.

kilimchoi avatar Aug 16 '24 18:08 kilimchoi

@pranay0318 did you train a lora using ai-toolkit? did you have to convert it using the script from the huggingface mentioned above?

kilimchoi avatar Aug 16 '24 19:08 kilimchoi

I was getting the same thing with SimpleTuner after some changes went in on the 13th. Looking at it, the keys didn't even seem to make sense to convert. I think they have claimed to have fixed the issue since then. I didn't read what caused the issue, but a bunch of dependency updates had gone in.

I trained a Lora with ai-tools over last night and one yesterday and I don't have this issue though.

markrmiller avatar Aug 16 '24 21:08 markrmiller

@markrmiller

ok training a new lora removed the error messages but lora doesn't seem to work as well on comfy. All the parameters are identical btw.

Image generated at the end of training: image

Image generated from the comfy workflow: image

kilimchoi avatar Aug 16 '24 23:08 kilimchoi

I've been using a 1.5 weight on the Lora in comfy. Which would normally be pretty darn high, but it makes a good image for my subject.

markrmiller avatar Aug 17 '24 04:08 markrmiller

Recent updates to comfy in the last day or two should resolve needing to increase strength by applying stochastic rounding when applying the lora to a 8bit model.

Regarding the images not matching exactly. the seeds function differently from comfy ui and ai-toolkit, so results will not be the same. Also, ai-toolkit does not merge the lora into an 8bit model when generating samples, which will have a different effect than merging them into an 8bit model on comfy.

jaretburkett avatar Aug 19 '24 03:08 jaretburkett

Im getting this same issue. Just trained a lora using ai-toolkit last night, and the samples turned out great. I downloaded the lora, put it into a simple comfy workflow that loads the lora, but I get those same warnings about not finding the lora_A.weight and lora_B.weight. I used the script to convert the safetensor but have the same issue.

I wrote out all the keys from the safetensor on the converted file to the terminal and I see all the ones its complaining about not being able to load though, so not sure whats going on.

sbonkosky avatar Aug 25 '24 14:08 sbonkosky