From 19a88f1aa4d22f91012dd644de105eaf06e41bb9 Mon Sep 17 00:00:00 2001 From: duhaifeng Date: Mon, 18 Nov 2024 18:15:10 +0800 Subject: [PATCH] feat: pass spare_params --- nodes.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nodes.py b/nodes.py index 28e0d08..28b7b19 100644 --- a/nodes.py +++ b/nodes.py @@ -235,7 +235,7 @@ class BiRefNet_Lite: if hasattr(model_class, "_no_split_modules"): delattr(model_class, "_no_split_modules") - birefnet = AutoModelForImageSegmentation.from_pretrained(local_model_path,trust_remote_code=True) + birefnet = AutoModelForImageSegmentation.from_pretrained(local_model_path,trust_remote_code=True, **spare_params) else: birefnet = AutoModelForImageSegmentation.from_pretrained( "ZhengPeng7/BiRefNet", trust_remote_code=True