fix: fix no attribute del error

rubi-du · Nov 18, 2024 · 272f5cd · 272f5cd
1 parent 0a82a7b
commit 272f5cd
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/js/BiRefNet_Lite.js b/js/BiRefNet_Lite.js
@@ -81,8 +81,8 @@ app.registerExtension({
 				return;
 			}
 
-			if(node.widgets_values[3] != undefined){
-				localModelPathWidget.value = node.widgets_values[3];
+			if(node.widgets_values[8] != undefined){
+				localModelPathWidget.value = node.widgets_values[8];
 			}
 
 			// 初始状态控制：根据 load_local_model 的值决定是否显示 local_model_path

diff --git a/nodes.py b/nodes.py
@@ -232,8 +232,11 @@ def background_remove(self,
                     spare_params['max_memory'] = {0: f"{max_gpu_size}GiB", "cpu": f"{cpu_size}GiB"}
                     setattr(model_class, '_no_split_modules', ["Decoder", "SwinTransformer"])
                 else:
-                    if hasattr(model_class, "_no_split_modules"):
-                        delattr(model_class, "_no_split_modules")
+                    try:
+                        if hasattr(model_class, "_no_split_modules"):
+                            delattr(model_class, "_no_split_modules")
+                    except Exception as e:
+                        print('No need to delete:', e)
 
                 birefnet = AutoModelForImageSegmentation.from_pretrained(local_model_path,trust_remote_code=True, **spare_params)
             else: