mirror of
https://github.com/vladmandic/automatic
synced 2026-09-19 17:24:32 +02:00
@@ -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
|
||||
|
||||
@@ -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'):
|
||||
|
||||
+2
-2
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user