Unverified Commit 6281c1bd authored by AUTOMATIC1111's avatar AUTOMATIC1111 Committed by GitHub
Browse files

Merge pull request #6299 from stysmmaker/feat/latent-upscale-modes 

Add more latent upscale modes
parents 77c3bc77 b2151b93

modules/processing.py

+1 −1
Original line number Diff line number Diff line
@@ -713,7 +713,7 @@ class StableDiffusionProcessingTxt2Img(StableDiffusionProcessing): 
            for i in range(samples.shape[0]):

                save_intermediate(samples, i)



            samples = torch.nn.functional.interpolate(samples, size=(target_height // opt_f, target_width // opt_f), mode=latent_scale_mode)

            samples = torch.nn.functional.interpolate(samples, size=(target_height // opt_f, target_width // opt_f), mode=latent_scale_mode["mode"], antialias=latent_scale_mode["antialias"])



            # Avoid making the inpainting conditioning unless necessary as

            # this does need some extra compute to decode / encode the image again.

modules/shared.py

+5 −2
Original line number Diff line number Diff line
@@ -564,8 +564,11 @@ if os.path.exists(config_filename): 


latent_upscale_default_mode = "Latent"

latent_upscale_modes = {

    "Latent": "bilinear",

    "Latent (nearest)": "nearest",

    "Latent": {"mode": "bilinear", "antialias": False},

    "Latent (antialiased)": {"mode": "bilinear", "antialias": True},

    "Latent (bicubic)": {"mode": "bicubic", "antialias": False},

    "Latent (bicubic antialiased)": {"mode": "bicubic", "antialias": True},

    "Latent (nearest)": {"mode": "nearest", "antialias": False},

}



sd_upscalers = []

CRA Git | Maintained and supported by SUSTech CRA and CCSE