Skip to content

S
Stable Diffusion Webui
Commit 731eb727 authored by Andray's avatar Andray
Browse files
Options

fix sd2 switching

parent a30b19dd
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
modules/sd_models_config.py
View file @ 731eb727
...@@ -31,7 +31,7 @@ def is_using_v_parameterization_for_sd2(state_dict): ...@@ -31,7 +31,7 @@ def is_using_v_parameterization_for_sd2(state_dict):
import ldm.modules.diffusionmodules.openaimodel import ldm.modules.diffusionmodules.openaimodel
device = devices.cpu device = devices.device
with sd_disable_initialization.DisableInitialization(): with sd_disable_initialization.DisableInitialization():
unet = ldm.modules.diffusionmodules.openaimodel.UNetModel( unet = ldm.modules.diffusionmodules.openaimodel.UNetModel(
......
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