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

fix sd2 switching

Merge pull request #16079 from light-and-ray/fix_sd2_switching
fix sd2 switching
477869c0 · AUTOMATIC1111 · GitHub · ffead92d · 731eb727 · 477869c0
Commit 477869c0 authored Jul 06, 2024 by AUTOMATIC1111 Committed by GitHub Jul 06, 2024
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

modules/sd_models_config.py modules/sd_models_config.py +1 -1

No files found.
--- a/modules/sd_models_config.py
+++ b/modules/sd_models_config.py
@@ -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(