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

requirements.txt

@@ -0,0 +1,16 @@
+torch==2.5.1
+torchvision
+numpy<2
+opencv-python
+timm
+scipy
+scikit-image
+kornia
+einops
+
+tqdm
+prettytable
+
+transformers
+huggingface-hub>0.25
+accelerate