add linfusion

Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2024-10-09 11:01:16 -04:00
parent f17fbe096d
commit 723ce74986
9 changed files with 270 additions and 12 deletions
+5
View File
@@ -1346,6 +1346,11 @@ def load_diffuser(checkpoint_info=None, already_loaded_state_dict=None, timer=No
sd_model = sd_models_compile.compile_diffusers(sd_model)
timer.record("compile")
if shared.opts.enable_linfusion:
from modules import linfusion
linfusion.apply(sd_model)
timer.record("linfusion")
except Exception as e:
shared.log.error(f"Load {op}: {e}")
errors.display(e, "Model")