Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-18 16:54:33 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix OpenVINO and NNCF Lora case

Browse Source
This commit is contained in:
Disty0
2024-03-14 23:37:24 +03:00
parent 9d1642fd0b
commit ec0c4f475a
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
extensions-builtin/Lora/networks.py
+2 -2
View File
@@ -229,10 +229,10 @@ def load_networks(names, te_multipliers=None, unet_multipliers=None, dyn_dims=No
if recompile_model:
shared.log.info("LoRA recompiling model")
backup_lora_model = shared.compiled_model_state.lora_model
if shared.opts.nncf_compress_weights and not (shared.opts.cuda_compile and shared.opts.cuda_compile_backend == "openvino_fx"):
shared.sd_model = sd_models_compile.nncf_compress_weights(shared.sd_model)
if shared.opts.cuda_compile:
shared.sd_model = sd_models_compile.compile_diffusers(shared.sd_model)
if shared.opts.nncf_compress_weights:
shared.sd_model = sd_models_compile.nncf_compress_weights(shared.sd_model)
shared.compiled_model_state.lora_model = backup_lora_model