ai-models/RMBG-2.0
2
0
Fork 0
mirror of https://www.modelscope.cn/AI-ModelScope/RMBG-2.0.git synced 2026-04-02 19:12:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update README.md

Browse Source
This commit is contained in:
Cherrytest
2025-09-22 07:12:51 +00:00
parent fdc5d45001
commit cf6aec8113
2 changed files with 4 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
birefnet.py
View File

@ -6,6 +6,7 @@ from transformers import PretrainedConfig
class Config(PretrainedConfig):
def __init__(self) -> None:
super().__init__()
# 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