From f4833132705de8af8f88bbd4bec6ffcb670c69ca Mon Sep 17 00:00:00 2001
From: zmwv823 <913720128@qq.com>
Date: Wed, 4 Sep 2024 22:30:50 +0800
Subject: [PATCH] =?UTF-8?q?cpu=5Foffload=E6=A8=A1=E5=BC=8F=E4=B8=8B?=
 =?UTF-8?q?=EF=BC=8C=E5=8A=A0=E5=85=A5=E4=B8=8A=E6=AC=A1=E5=BF=98=E8=AE=B0?=
 =?UTF-8?q?=E7=9A=84=E6=9C=AC=E5=9C=B0=E6=A8=A1=E5=9E=8B=E5=88=87=E6=8D=A2?=
 =?UTF-8?q?=E3=80=82?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 nodes.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/nodes.py b/nodes.py
index 43ae3a0..a0122d7 100644
--- a/nodes.py
+++ b/nodes.py
@@ -120,7 +120,7 @@ class BiRefNet_Hugo:
                 "dtype": (["auto","fp16","bf16","fp32", "fp8_e4m3fn", "fp8_e4m3fnuz", "fp8_e5m2", "fp8_e5m2fnuz"],{"default":"fp32"}),
                 "cpu_offload": ("BOOLEAN", {"default": False, "label_on": "model_to_cpu", "label_off": "unload_model"}),
                 "Auto_Download_Path": ("BOOLEAN", {"default": True, "label_on": "rembg_local本地", "label_off": ".cache缓存"}),
-                "Show_Colors_In_Ternimal": ("BOOLEAN", {"default": False, "label_on": "show", "label_off": "hide"}),
+                "Show_Colors_In_Ternimal": ("BOOLEAN", {"default": False, "label_on": "yes", "label_off": "no"}),
             }
         }
 
@@ -164,7 +164,7 @@ class BiRefNet_Hugo:
         device = get_device_by_name(device)
         dtype = get_dtype_by_name(dtype)
             
-        if self.loaded_model_name != model_name:
+        if self.loaded_model_name != model:
             del self.model
             self.model = None
         if self.model == None:
@@ -209,7 +209,7 @@ class BiRefNet_Hugo:
         new_masks = torch.cat(processed_masks, dim=0)
         if cpu_offload == True:
             self.model.to("cpu")
-            self.loaded_model_name = model_name
+            self.loaded_model_name = model
         else:
             del self.model
             self.model = None