fix GetColorToneV2 load birefnet error

2024-09-15 23:00:50 +08:00
parent 15d3bb1fd9
commit 01a28f0eb1
1 changed files with 1 additions and 1 deletions
@@ -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):