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

fix sd2 switching
2024-07-06 10:41:16 +03:00 · 2024-07-06 10:41:16 +03:00 · 477869c044
parent ffead92d4e 731eb72774
commit 477869c044
1 changed files with 1 additions and 1 deletions
--- 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(