VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-09 06:45:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix SD1.5 LoRA extraction #1490

Browse Source
This commit is contained in:
Kohya S
2024-08-22 21:33:15 +09:00
parent 74f91c2ff7
commit afb971f9c3
1 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
networks/lora.py
View File

@@ -815,7 +815,8 @@ def create_network_from_weights(multiplier, file, vae, text_encoder, unet, weigh
weights_sd = torch.load(file, map_location="cpu") weights_sd = torch.load(file, map_location="cpu")
# if keys are Diffusers based, convert to SAI based # if keys are Diffusers based, convert to SAI based
convert_diffusers_to_sai_if_needed(weights_sd) if is_sdxl:
convert_diffusers_to_sai_if_needed(weights_sd)
# get dim/alpha mapping # get dim/alpha mapping
modules_dim = {} modules_dim = {}
@@ -840,7 +841,13 @@ def create_network_from_weights(multiplier, file, vae, text_encoder, unet, weigh
module_class = LoRAInfModule if for_inference else LoRAModule module_class = LoRAInfModule if for_inference else LoRAModule
network = LoRANetwork( network = LoRANetwork(
text_encoder, unet, multiplier=multiplier, modules_dim=modules_dim, modules_alpha=modules_alpha, module_class=module_class text_encoder,
unet,
multiplier=multiplier,
modules_dim=modules_dim,
modules_alpha=modules_alpha,
module_class=module_class,
is_sdxl=is_sdxl,
) )
# block lr # block lr