add support for SDXL loras with te1/te2 modules

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

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