diff --git a/modules/lora/extra_networks_lora.py b/modules/lora/extra_networks_lora.py index faccc7d6e..a17914683 100644 --- a/modules/lora/extra_networks_lora.py +++ b/modules/lora/extra_networks_lora.py @@ -4,7 +4,7 @@ import re import numpy as np from modules.lora import networks, lora_overrides, lora_load from modules.lora import lora_common as l -from modules import extra_networks, shared +from modules import extra_networks, shared, sd_models debug = os.environ.get('SD_LORA_DEBUG', None) is not None @@ -179,6 +179,7 @@ class ExtraNetworkLora(extra_networks.ExtraNetwork): job = shared.state.job shared.state.job = 'LoRA' lora_load.network_load(names, te_multipliers, unet_multipliers, dyn_dims) # load only on first call + sd_models.set_diffuser_offload(shared.sd_model, op="model") shared.state.job = job else: lora_load.network_load(names, te_multipliers, unet_multipliers, dyn_dims) # load diff --git a/modules/teacache/__init__.py b/modules/teacache/__init__.py index cb06c1f26..89e3cc0c8 100644 --- a/modules/teacache/__init__.py +++ b/modules/teacache/__init__.py @@ -9,6 +9,8 @@ supported_models = ['Flux', 'CogVideoX', 'Mochi', 'LTX'] def apply_teacache(p): from modules import shared + if not shared.opts.teacache_enabled: + return if not any(shared.sd_model.__class__.__name__.startswith(x) for x in supported_models): return if not hasattr(shared.sd_model, 'transformer'): diff --git a/requirements.txt b/requirements.txt index f1de3abd7..df120f3fa 100644 --- a/requirements.txt +++ b/requirements.txt @@ -34,7 +34,7 @@ pi-heif rich==13.9.4 safetensors==0.5.3 tensordict==0.1.2 -peft==0.14.0 +peft==0.15.1 httpx==0.24.1 compel==2.0.3 torchsde==0.2.6 @@ -52,7 +52,7 @@ numba==0.59.1 protobuf==4.25.3 pytorch_lightning==1.9.4 tokenizers==0.21.1 -transformers==4.50.3 +transformers==4.51.0 urllib3==1.26.19 Pillow==10.4.0 timm==0.9.16