better diffuser patcher

2026-03-03 12:09:51 +00:00 · 2024-02-02 00:32:15 -08:00
parent cb1aa493e2
commit b944aa9242
2 changed files with 8 additions and 4 deletions
--- a/modules_forge/diffusers_patcher.py
+++ b/modules_forge/diffusers_patcher.py
@@ -43,6 +43,6 @@ class DiffusersModelPatcher:
        area = 2 * batchsize * latent_width * latent_height
        inference_memory = (((area * 0.6) / 0.9) + 1024) * (1024 * 1024)
        model_management.load_models_gpu(
-            models=[self.pipeline],
+            models=[self.patcher],
            memory_required=inference_memory
        )