AI/RMBG-2.0
1
0
Fork 0
mirror of https://www.modelscope.cn/AI-ModelScope/RMBG-2.0.git synced 2025-04-10 11:50:22 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update birefnet.py

Browse Source
This commit is contained in:
ai-modelscope 2025-02-20 06:29:05 +08:00
parent b104609356
commit 839b3a2e36
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
birefnet.py
View File

@ -2,9 +2,9 @@
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