Fix the wrong bb setting. Fix the bug by latest transformers and update req.txt.

birefnet.py

@@ -2,9 +2,10 @@
 
 import os
 import math
+from transformers import PretrainedConfig
 
 
-class Config():
+class Config(PretrainedConfig):
     def __init__(self) -> None:
         # PATH settings
         self.sys_home_dir = os.path.expanduser('~')     # Make up your file system as: SYS_HOME_DIR/codes/dis/BiRefNet, SYS_HOME_DIR/datasets/dis/xx, SYS_HOME_DIR/weights/xx

handler.py

@@ -67,6 +67,7 @@ usage_to_weights_file = {
     'General-Lite-2K': 'BiRefNet_lite-2K',
     'General-reso_512': 'BiRefNet-reso_512',
     'Matting': 'BiRefNet-matting',
+    'Matting-HR': 'BiRefNet_HR-Matting',
     'Portrait': 'BiRefNet-portrait',
     'DIS': 'BiRefNet-DIS5K',
     'HRSOD': 'BiRefNet-HRSOD',
@@ -83,7 +84,7 @@ if usage in ['General-Lite-2K']:
     resolution = (2560, 1440)
 elif usage in ['General-reso_512']:
     resolution = (512, 512)
-elif usage in ['General-HR']:
+elif usage in ['General-HR', 'Matting-HR']:
     resolution = (2048, 2048)
 else:
     resolution = (1024, 1024) 

requirements.txt

@@ -1,7 +1,5 @@
---extra-index-url https://download.pytorch.org/whl/cu118
-torch==2.0.1
---extra-index-url https://download.pytorch.org/whl/cu118
-torchvision==0.15.2
+torch==2.5.1
+torchvision
 numpy<2
 opencv-python
 timm