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

@@ -1,5 +1,5 @@
 torch==2.5.1
-torchvision==0.20.1
+torchvision
 numpy<2
 opencv-python
 timm