Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-09-19 19:05:28 +00:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

add_loras_to_unet: add preprocess values as exclusions in last step

Browse source
This commit is contained in:
Pierre Chapuis 2024-03-05 19:29:25 +01:00
parent 4259261f17
commit fb90b00e75
1 changed files with 10 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -143,11 +143,17 @@ class SDLoraManager:
loras_excluded = {k: v for k, v in unet_loras.items() if any(x in k for x in preprocess.keys())} loras_excluded = {k: v for k, v in unet_loras.items() if any(x in k for x in preprocess.keys())}
loras_remaining = {k: v for k, v in unet_loras.items() if k not in loras_excluded} loras_remaining = {k: v for k, v in unet_loras.items() if k not in loras_excluded}
for 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 in k} ls = {k: v for k, v in loras_excluded.items() if exc_k in k}
auto_attach_loras(ls, self.unet, include=[v], debug_map=debug_map) auto_attach_loras(ls, self.unet, include=[exc_v], debug_map=debug_map)
auto_attach_loras(loras_remaining, self.unet, exclude=exclude, include=include, debug_map=debug_map) auto_attach_loras(
loras_remaining,
self.unet,
exclude=[*exclude, *preprocess.values()],
include=include,
debug_map=debug_map,
)
def remove_loras(self, *names: str) -> None: def remove_loras(self, *names: str) -> None:
"""Remove multiple LoRAs from the target. """Remove multiple LoRAs from the target.