jprdonnelly/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-12-19 17:47:25 -05:00
Code Issues Packages Projects Releases Wiki Activity

fix sd2 switching

Browse Source
This commit is contained in:
Andray
2024-06-23 21:16:48 +04:00
parent a30b19dd55
commit 731eb72774
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
modules/sd_models_config.py
View File

@@ -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(