From b31f81628f70e725197e9b5937baa4e121c3adb3 Mon Sep 17 00:00:00 2001 From: layerdiffusion <19834515+lllyasviel@users.noreply.github.com> Date: Wed, 14 Aug 2024 20:39:00 -0700 Subject: [PATCH] Revert "simplify codes" This reverts commit 2cc5aa7a3e7e3418e3aed3440a945f43b338cfc6. --- backend/memory_management.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/backend/memory_management.py b/backend/memory_management.py index 4f64c61a..86e9f09f 100644 --- a/backend/memory_management.py +++ b/backend/memory_management.py @@ -375,7 +375,7 @@ class LoadedModel: self.model.model_patches_to(self.model.model_dtype()) try: - self.real_model = self.model.forge_patch_model(patch_model_to) + self.real_model = self.model.forge_patch_model(device_to=patch_model_to) except Exception as e: self.model.forge_unpatch_model(self.model.offload_device) self.model_unload()