From 3e20bbbedcb00fa9a4306c24ae99a5f595717597 Mon Sep 17 00:00:00 2001
From: DeepBeepMeep <deepbeepmeep@yahoo.com>
Date: Wed, 7 May 2025 02:52:38 +0200
Subject: [PATCH] More VRAM optimisations

---
 requirements.txt | 2 +-
 wgp.py           | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index ec8ea94..5327d4b 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -16,7 +16,7 @@ gradio==5.23.0
 numpy>=1.23.5,<2
 einops
 moviepy==1.0.3
-mmgp==3.4.3
+mmgp==3.4.4
 peft==0.14.0
 mutagen
 pydantic==2.10.6
diff --git a/wgp.py b/wgp.py
index 5357adb..c02161b 100644
--- a/wgp.py
+++ b/wgp.py
@@ -40,7 +40,7 @@ global_queue_ref = []
 AUTOSAVE_FILENAME = "queue.zip"
 PROMPT_VARS_MAX = 10
 
-target_mmgp_version = "3.4.3"
+target_mmgp_version = "3.4.4"
 from importlib.metadata import version
 mmgp_version = version("mmgp")
 if mmgp_version != target_mmgp_version:
@@ -1922,7 +1922,7 @@ def load_models(model_filename):
     wan_model._model_file_name = new_transformer_filename
     kwargs = { "extraModelsToQuantize": None}   
     if profile == 2 or profile == 4:
-        kwargs["budgets"] = { "transformer" : 100 if preload  == 0 else preload, "text_encoder" : 100, "*" : 1000 }
+        kwargs["budgets"] = { "transformer" : 100 if preload  == 0 else preload, "text_encoder" : 100 if preload  == 0 else preload, "*" : 1000 }
         # if profile == 4:
         #     kwargs["partialPinning"] = True
     elif profile == 3: