From 4bea93bc06c8c65f7ab93769ac4c3889ff3b9c72 Mon Sep 17 00:00:00 2001 From: Won-Kyu Park Date: Fri, 20 Sep 2024 08:27:58 +0900 Subject: [PATCH] fixed typo in the flux lora map --- modules/models/flux/flux.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules/models/flux/flux.py b/modules/models/flux/flux.py index 42e9ea788..109c72904 100644 --- a/modules/models/flux/flux.py +++ b/modules/models/flux/flux.py @@ -360,4 +360,4 @@ class FLUX1Inferencer(torch.nn.Module): yield f"transformer.single_transformer_blocks.{i}.proj_mlp", f"diffusion_model_single_blocks_{i}_linear1_mlp_proj" yield f"transformer.single_transformer_blocks.{i}.proj_out", f"diffusion_model_single_blocks_{i}_linear2" - yield f"transformer.single_transformer_blocks.{i}.morm.linear", f"diffusion_model_single_blocks_{i}_modulation_lin" + yield f"transformer.single_transformer_blocks.{i}.norm.linear", f"diffusion_model_single_blocks_{i}_modulation_lin"