From 839b3a2e36c0286773f1aa84b46bc78c254f2ac4 Mon Sep 17 00:00:00 2001 From: ai-modelscope Date: Thu, 20 Feb 2025 06:29:05 +0800 Subject: [PATCH] Update birefnet.py --- birefnet.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/birefnet.py b/birefnet.py index 1ed28de..d68d7d3 100644 --- a/birefnet.py +++ b/birefnet.py @@ -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