fixed typo in the flux lora map

This commit is contained in:
Won-Kyu Park 2024-09-20 08:27:58 +09:00
parent 28eca46959
commit 4bea93bc06
No known key found for this signature in database
GPG Key ID: 53AA79C8C9535D15

View File

@ -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"