diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-01-04 13:47:36 +0300 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-01-04 13:47:36 +0300 |
| commit | 6281c1bdb44165ef466bbfdc1d7490e33269bfdc (patch) | |
| tree | 4650be258ee96c8ad0cc0816371805342f66e121 /modules | |
| parent | 77c3bc77479cb41be1fbae310d073da4b43530c1 (diff) | |
| parent | b2151b934fe0a3613570c6abd7615d3788fd1c8f (diff) | |
Merge pull request #6299 from stysmmaker/feat/latent-upscale-modes
Add more latent upscale modes
Diffstat (limited to 'modules')
| -rw-r--r-- | modules/processing.py | 2 | ||||
| -rw-r--r-- | modules/shared.py | 7 |
2 files changed, 6 insertions, 3 deletions
diff --git a/modules/processing.py b/modules/processing.py index 93e75ba6..fd7c7015 100644 --- a/modules/processing.py +++ b/modules/processing.py @@ -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.
diff --git a/modules/shared.py b/modules/shared.py index 7588c47b..c1b20081 100644 --- a/modules/shared.py +++ b/modules/shared.py @@ -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 = []
|