From 43d265faf4e364d0a53a24a2deb183c23cf61e16 Mon Sep 17 00:00:00 2001 From: TSC <112517630+LucianoCirino@users.noreply.github.com> Date: Mon, 18 Sep 2023 21:28:10 -0500 Subject: [PATCH] Few more I missed --- efficiency_nodes.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/efficiency_nodes.py b/efficiency_nodes.py index be5d20b..0ef9eaf 100644 --- a/efficiency_nodes.py +++ b/efficiency_nodes.py @@ -1043,7 +1043,7 @@ class TSC_KSampler: elif X_type == "Refiner": ckpt_dict = [] lora_dict = [] - elif X_type == "LoRA": + elif X_type in ("LoRA", "LoRA Stacks"): ckpt_dict = [] refn_dict = [] @@ -1572,7 +1572,7 @@ class TSC_KSampler: clear_cache_by_exception(xyplot_id, lora_dict=[], refn_dict=[]) elif X_type == "Refiner": clear_cache_by_exception(xyplot_id, ckpt_dict=[], lora_dict=[]) - elif X_type == "LoRA": + elif X_type in ("LoRA", "LoRA Stacks"): clear_cache_by_exception(xyplot_id, ckpt_dict=[], refn_dict=[]) # __________________________________________________________________________________________________________