From 2cc5aa7a3e7e3418e3aed3440a945f43b338cfc6 Mon Sep 17 00:00:00 2001 From: layerdiffusion <19834515+lllyasviel@users.noreply.github.com> Date: Wed, 14 Aug 2024 20:35:28 -0700 Subject: [PATCH] simplify codes --- 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 86e9f09f..4f64c61a 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(device_to=patch_model_to) + self.real_model = self.model.forge_patch_model(patch_model_to) except Exception as e: self.model.forge_unpatch_model(self.model.offload_device) self.model_unload()