Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-11-22 06:08:46 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix text encoder LoRAs

Browse source
This commit is contained in:
Pierre Chapuis 2023-09-04 14:38:03 +02:00
parent ebfa51f662
commit 566656a539
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

@ -100,7 +100,7 @@ class SD1LoraAdapter(fl.Chain, Adapter[StableDiffusion_1]):
for model_name in MODELS: for model_name in MODELS:
if not (model_targets := sub_targets.get(model_name, [])): if not (model_targets := sub_targets.get(model_name, [])):
continue continue
model = getattr(target, model_name) model = getattr(target, "clip_text_encoder" if model_name == "text_encoder" else model_name)
if model.find(SingleLoraAdapter): if model.find(SingleLoraAdapter):
raise NotImplementedError(f"{model} already contains LoRA layers") raise NotImplementedError(f"{model} already contains LoRA layers")