From 01a28f0eb1de44d61b8b4f99f750fbb28072328b Mon Sep 17 00:00:00 2001
From: chflame163 <chflame@163.com>
Date: Sun, 15 Sep 2024 23:00:50 +0800
Subject: [PATCH] fix GetColorToneV2 load birefnet error

---
 py/get_color_tone_v2.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/py/get_color_tone_v2.py b/py/get_color_tone_v2.py
index 630548b..79e47b4 100644
--- a/py/get_color_tone_v2.py
+++ b/py/get_color_tone_v2.py
@@ -61,7 +61,7 @@ class GetColorToneV2:
         max_batch = max(len(_images), len(_masks))
 
         if remove_bkgd_method == 'BiRefNet':
-            from .birefnet_func import BiRefNetRemoveBackground
+            from .birefnet_legacy import BiRefNetRemoveBackground
             birefnetrmbg = BiRefNetRemoveBackground()
 
         for i in range(max_batch):