From 922b8d2ccbd79eccc1165c230dc15beff5e6406b Mon Sep 17 00:00:00 2001 From: DeepBeepMeep Date: Mon, 24 Mar 2025 12:50:18 +0100 Subject: [PATCH] fixed memory leak while switching tabs --- gradio_server.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/gradio_server.py b/gradio_server.py index 5510b9c..b841907 100644 --- a/gradio_server.py +++ b/gradio_server.py @@ -658,8 +658,8 @@ def load_models(i2v): kwargs = { "extraModelsToQuantize": None} if profile == 2 or profile == 4: kwargs["budgets"] = { "transformer" : 100 if preload == 0 else preload, "text_encoder" : 100, "*" : 1000 } - if profile == 4: - kwargs["partialPinning"] = True + # if profile == 4: + # kwargs["partialPinning"] = True elif profile == 3: kwargs["budgets"] = { "*" : "70%" } offloadobj = offload.profile(pipe, profile_no= profile, compile = compile, quantizeTransformer = quantizeTransformer, loras = "transformer", **kwargs)