add XL support for live previews: approx and TAESD

This commit is contained in:
AUTOMATIC1111
2023-07-13 17:24:54 +03:00
parent 6f23da603d
commit b8159d0919
3 changed files with 40 additions and 25 deletions

View File

@@ -48,7 +48,7 @@ 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_xl = True
model.is_sdxl = True
sgm.models.diffusion.DiffusionEngine.get_learned_conditioning = get_learned_conditioning