fix sd2 switching

2024-06-23 21:16:48 +04:00 · 2024-06-23 21:16:48 +04:00 · 731eb72774
parent a30b19dd55
commit 731eb72774
1 changed files with 1 additions and 1 deletions
--- a/modules/sd_models_config.py
+++ b/modules/sd_models_config.py
@ -31,7 +31,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(