Apply requested changes

Co-Authored-By: psychedelicious <4822129+psychedelicious@users.noreply.github.com>
This commit is contained in:
Sergey Borisov
2023-08-10 06:19:22 +03:00
parent 17fed1c870
commit e9ec5ab85c
4 changed files with 40 additions and 19 deletions

View File

@ -58,6 +58,8 @@ def stable_diffusion_step_callback(
# TODO: only output a preview image when requested
if base_model in [BaseModelType.StableDiffusionXL, BaseModelType.StableDiffusionXLRefiner]:
# fast latents preview matrix for sdxl
# generated by @StAlKeR7779
sdxl_latent_rgb_factors = torch.tensor(
[
# R G B
@ -72,9 +74,6 @@ def stable_diffusion_step_callback(
sdxl_smooth_matrix = torch.tensor(
[
# [ 0.0478, 0.1285, 0.0478],
# [ 0.1285, 0.2948, 0.1285],
# [ 0.0478, 0.1285, 0.0478],
[0.0358, 0.0964, 0.0358],
[0.0964, 0.4711, 0.0964],
[0.0358, 0.0964, 0.0358],