Skip to content

S
Stable Diffusion Webui
Commit 47f1d42a authored by huchenlei's avatar huchenlei
Browse files
Options

Fix for SD15 models

parent 58eec83a
Show whitespace changes
Inline Side-by-side
Showing with 5 additions and 2 deletions
modules/sd_models.py
View file @ 47f1d42a
...@@ -553,8 +553,11 @@ def repair_config(sd_config): ...@@ -553,8 +553,11 @@ def repair_config(sd_config):
# Do not use checkpoint for inference. # Do not use checkpoint for inference.
# This helps prevent extra performance overhead on checking parameters. # This helps prevent extra performance overhead on checking parameters.
# The perf overhead is about 100ms/it on 4090. # The perf overhead is about 100ms/it on 4090 for SDXL.
if hasattr(sd_config.model.params, "network_config"):
sd_config.model.params.network_config.params.use_checkpoint = False sd_config.model.params.network_config.params.use_checkpoint = False
if hasattr(sd_config.model.params, "unet_config"):
sd_config.model.params.unet_config.params.use_checkpoint = False
def rescale_zero_terminal_snr_abar(alphas_cumprod): def rescale_zero_terminal_snr_abar(alphas_cumprod):
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment