Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-11-09 23:12:02 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

LoRA loading: forward exclusions when preprocessing parts of the UNet

Browse source
This commit is contained in:
Pierre Chapuis 2024-03-13 14:58:26 +01:00
parent ff5341c85c
commit e32d8d16f0
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
src/refiners/foundationals/latent_diffusion/lora.py
View file

@ -184,7 +184,7 @@ class SDLoraManager:
for exc_k, exc_v in preprocess.items(): for exc_k, exc_v in preprocess.items():
ls = {k: v for k, v in loras_excluded.items() if exc_k in k} ls = {k: v for k, v in loras_excluded.items() if exc_k in k}
auto_attach_loras(ls, self.unet, include=[exc_v], debug_map=debug_map) auto_attach_loras(ls, self.unet, include=[exc_v], exclude=exclude, debug_map=debug_map)
auto_attach_loras( auto_attach_loras(
loras_remaining, loras_remaining,