mirror/stable-diffusion-webui
2
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-02-11 15:41:06 +08:00
Code Issues 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
commit 477869c044
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats 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(