mirrors/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-09-13 04:55:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix loras not working when loading checkpoint with config.

Browse Source
This commit is contained in:
comfyanonymous
2023-07-05 19:37:19 -04:00
parent ccb1b25908
commit 603f02d613
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
comfy/utils.py
View File

@@ -140,8 +140,10 @@ def unet_to_diffusers(unet_config):
channel_mult = unet_config["channel_mult"] channel_mult = unet_config["channel_mult"]
transformer_depth = unet_config["transformer_depth"] transformer_depth = unet_config["transformer_depth"]
num_blocks = len(channel_mult) num_blocks = len(channel_mult)
if not isinstance(num_res_blocks, list): if isinstance(num_res_blocks, int):
num_res_blocks = [num_res_blocks] * num_blocks num_res_blocks = [num_res_blocks] * num_blocks
if isinstance(transformer_depth, int):
transformer_depth = [transformer_depth] * num_blocks
transformers_per_layer = [] transformers_per_layer = []
res = 1 res = 1
@@ -152,7 +154,7 @@ def unet_to_diffusers(unet_config):
transformers_per_layer.append(transformers) transformers_per_layer.append(transformers)
res *= 2 res *= 2
transformers_mid = unet_config.get("transformer_depth_middle", transformers_per_layer[-1]) transformers_mid = unet_config.get("transformer_depth_middle", transformer_depth[-1])
diffusers_unet_map = {} diffusers_unet_map = {}
for x in range(num_blocks): for x in range(num_blocks):