OriLib commited on
Commit
e034044
·
verified ·
1 Parent(s): 8d8654b

Update birefnet.py

Browse files
Files changed (1) hide show
  1. birefnet.py +2 -2
birefnet.py CHANGED
@@ -2,9 +2,9 @@
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
+ class Config(PretrainedConfig):
 
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