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

fix sd2 switching
AUTOMATIC1111 · Jul 6, 2024 · 477869c · 477869c
2 parents ffead92 + 731eb72
commit 477869c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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(