Add torch.device

2026-09-20 01:31:13 +02:00 · 2023-11-17 20:35:06 -06:00
parent abb4739590
commit cb8be0fc6e
3 changed files with 23 additions and 26 deletions
@@ -311,7 +311,7 @@ def run_MEHmodelmerger(id_task, **kwargs):  # pylint: disable=unused-argument
            kwargs.pop("beta_mid_block", None)
            kwargs.pop("beta_out_blocks", None)
            kwargs.pop("beta_preset", None)
-            
+
    if kwargs["device"] == "cuda":
        sd_models.unload_model_weights()