1580158
/
sd-webui-lora-block-weight
mirror of https://github.com/hako-mikan/sd-webui-lora-block-weight.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

#140

ver11
hako-mikan 2023-12-18 23:19:10 +09:00 committed by GitHub
parent b19aa6cc58
commit 885db95104
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
scripts/lora_block_weight.py
View File

@ -384,9 +384,10 @@ class Script(modules.scripts.Script):
emb_db = sd_hijack.model_hijack.embedding_db
for net in lora.loaded_loras:
for emb_name, embedding in net.bundle_embeddings.items():
if embedding.loaded:
emb_db.register_embedding_by_name(None, shared.sd_model, emb_name)
if hasattr(net,"bundle_embeddings"):
for emb_name, embedding in net.bundle_embeddings.items():
if embedding.loaded:
emb_db.register_embedding_by_name(None, shared.sd_model, emb_name)
lora.loaded_loras.clear()