ZhengPeng7 commited on
Commit
0e81815
1 Parent(s): 407e35d

For users to load in one key.

Browse files
Files changed (1) hide show
  1. hf_config.py +9 -0
hf_config.py ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ from transformers import PretrainedConfig
2
+
3
+ class BiRefNetConfig(PretrainedConfig):
4
+ model_type = "SegformerForSemanticSegmentation"
5
+ def __init__(
6
+ self,
7
+ **kwargs
8
+ ):
9
+ super().__init__(**kwargs)