wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-24 21:30:15 +08:00
Code Issues Actions 12 Packages Projects Releases Wiki Activity

remove debug message

Browse Source
This commit is contained in:
Dr.Lt.Data 2023-03-28 22:39:49 +09:00
parent bd0a8163eb
commit 5239e7d508
1 changed files with 1 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
comfy/sd.py
View File

@ -285,6 +285,7 @@ class ModelPatcher:
model_sd = self.model.state_dict() model_sd = self.model.state_dict()
for k in patches: for k in patches:
if k in model_sd: if k in model_sd:
# block_key pattern is (input_blocks|output_blocks|middle).(block number)
sk = k.split(".") sk = k.split(".")
block_key = ".".join(sk[2:4]) block_key = ".".join(sk[2:4])
if block_weights.__contains__(block_key): if block_weights.__contains__(block_key):
@ -315,13 +316,6 @@ class ModelPatcher:
alpha = p[k][0] alpha = p[k][0]
if key.startswith("model.diffusion_model."):
print(f"{key}: {alpha}")
# sk = key.split(".")
# block_key = ".".join(sk[2:4])
# if LORA_BLOCK_WEIGHTS.__contains__(block_key):
# alpha *= LORA_BLOCK_WEIGHTS[block_key]
if len(v) == 4: #lora/locon if len(v) == 4: #lora/locon
mat1 = v[0] mat1 = v[0]
mat2 = v[1] mat2 = v[1]