hoelee/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-08-09 13:49:48 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

add support for SDXL loras with te1/te2 modules

Browse Source
This commit is contained in:
AUTOMATIC1111
2023-07-13 21:17:50 +03:00
parent ff73841c60
commit 6c5f83b19b
3 changed files with 33 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
modules/sd_models.py
View File

@@ -289,7 +289,8 @@ def load_model_weights(model, checkpoint_info: CheckpointInfo, state_dict, timer
if state_dict is None:
state_dict = get_checkpoint_state_dict(checkpoint_info, timer)
if hasattr(model, 'conditioner'):
model.is_sdxl = hasattr(model, 'conditioner')
if model.is_sdxl:
sd_models_xl.extend_sdxl(model)
model.load_state_dict(state_dict, strict=False)

1
modules/sd_models_xl.py
View File

@@ -48,7 +48,6 @@ def extend_sdxl(model):
discretization = sgm.modules.diffusionmodules.discretizer.LegacyDDPMDiscretization()
model.alphas_cumprod = torch.asarray(discretization.alphas_cumprod, device=devices.device, dtype=dtype)
model.is_sdxl = True
sgm.models.diffusion.DiffusionEngine.get_learned_conditioning = get_learned_conditioning