hoelee/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-08-03 10:50:23 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #16079 from light-and-ray/fix_sd2_switching

Browse Source 
fix sd2 switching
This commit is contained in:
AUTOMATIC1111
2024-07-06 10:41:16 +03:00
committed by GitHub
parent ffead92d4e 731eb72774
commit 477869c044
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

@@ -33,7 +33,7 @@ def is_using_v_parameterization_for_sd2(state_dict):
import ldm.modules.diffusionmodules.openaimodel
device = devices.cpu
device = devices.device
with sd_disable_initialization.DisableInitialization():
unet = ldm.modules.diffusionmodules.openaimodel.UNetModel(