ZhengPeng7 commited on
Commit
6d0fceb
·
1 Parent(s): bdff8f5

Fix the bug by latest transformers and update req.txt.

Browse files
Files changed (2) hide show
  1. birefnet.py +2 -1
  2. requirements.txt +2 -4
birefnet.py CHANGED
@@ -2,9 +2,10 @@
2
 
3
  import os
4
  import math
 
5
 
6
 
7
- class Config():
8
  def __init__(self) -> None:
9
  # PATH settings
10
  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
 
2
 
3
  import os
4
  import math
5
+ from transformers import PretrainedConfig
6
 
7
 
8
+ class Config(PretrainedConfig):
9
  def __init__(self) -> None:
10
  # PATH settings
11
  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 CHANGED
@@ -1,7 +1,5 @@
1
- --extra-index-url https://download.pytorch.org/whl/cu118
2
- torch==2.0.1
3
- --extra-index-url https://download.pytorch.org/whl/cu118
4
- torchvision==0.15.2
5
  numpy<2
6
  opencv-python
7
  timm
 
1
+ torch==2.5.1
2
+ torchvision
 
 
3
  numpy<2
4
  opencv-python
5
  timm