not-lain commited on
Commit
7ca6a91
·
verified ·
1 Parent(s): 57558e2

Push model using huggingface_hub.

Browse files
Files changed (3) hide show
  1. README.md +35 -0
  2. config.json +24 -0
  3. model.safetensors +3 -0
README.md ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: nobg
3
+ tags:
4
+ - birefnet
5
+ - model_hub_mixin
6
+ - nobg
7
+ - pytorch_model_hub_mixin
8
+ ---
9
+
10
+ <p align="center">
11
+ <img src="https://usefeyn.com/feyn/feyn_mark.svg"/>
12
+ </p>
13
+
14
+ This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration.
15
+
16
+ Library: [nobg](https://github.com/feyninc/nobg)
17
+
18
+ ## how to load
19
+ ```
20
+ pip install nobg
21
+ ```
22
+
23
+ use the AutoModel class
24
+ ```python
25
+ from nobg import AutoModel
26
+ model = AutoModel.from_pretrained("nobg/BiRefNet-Finetuned")
27
+ ```
28
+ or you can use the model class directly
29
+ ```python
30
+ from nobg import BiRefNet
31
+ model = BiRefNet.from_pretrained("nobg/BiRefNet-Finetuned")
32
+ ```
33
+
34
+ ## Contributions
35
+ Any contributions are welcome at https://github.com/feyninc/nobg
config.json ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dec_channels_inter": 64,
3
+ "depths": [
4
+ 2,
5
+ 2,
6
+ 18,
7
+ 2
8
+ ],
9
+ "drop_path_rate": 0.2,
10
+ "embed_dim": 192,
11
+ "image_size": 1024,
12
+ "mlp_ratio": 4.0,
13
+ "num_heads": [
14
+ 6,
15
+ 12,
16
+ 24,
17
+ 48
18
+ ],
19
+ "patch_size": 4,
20
+ "use_gradient_attention": true,
21
+ "use_image_patch_injection": true,
22
+ "use_multi_scale_input": true,
23
+ "window_size": 12
24
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:88946d381613506953012c4e03a44aba5ce4bc818aa8763aa4c6069db0ddb52c
3
+ size 884878824