ZhengPeng7 commited on
Commit
80b33a4
1 Parent(s): b6ce8c8

Rename BiRefNet_T to BiRefNet_lite.

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -63,11 +63,11 @@ pip install -qr https://raw.githubusercontent.com/ZhengPeng7/BiRefNet/main/requi
63
  ```python
64
  # Load BiRefNet with weights
65
  from transformers import AutoModelForImageSegmentation
66
- birefnet = AutoModelForImageSegmentation.from_pretrained('zhengpeng7/BiRefNet_T', trust_remote_code=True)
67
  ```
68
 
69
  #### Use codes from GitHub + weights from HuggingFace
70
- > Only use the weights on HuggingFace -- Pro: codes are always latest; Con: Need to clone the BiRefNet repo from my GitHub.
71
 
72
  ```shell
73
  # Download codes
@@ -81,7 +81,7 @@ from models.birefnet import BiRefNet
81
 
82
  # Load weights from Hugging Face Models
83
  ### >>> Remember to set the `bb` in `config.py` as `swin_v1_t` to use this tiny version. <<< ###
84
- birefnet = BiRefNet.from_pretrained('zhengpeng7/BiRefNet_T')
85
  ```
86
 
87
  #### Use codes from GitHub + weights from HuggingFace
 
63
  ```python
64
  # Load BiRefNet with weights
65
  from transformers import AutoModelForImageSegmentation
66
+ birefnet = AutoModelForImageSegmentation.from_pretrained('zhengpeng7/BiRefNet_lite', trust_remote_code=True)
67
  ```
68
 
69
  #### Use codes from GitHub + weights from HuggingFace
70
+ > Only use the weights on HuggingFace -- Pro: codes are always the latest; Con: Need to clone the BiRefNet repo from my GitHub.
71
 
72
  ```shell
73
  # Download codes
 
81
 
82
  # Load weights from Hugging Face Models
83
  ### >>> Remember to set the `bb` in `config.py` as `swin_v1_t` to use this tiny version. <<< ###
84
+ birefnet = BiRefNet.from_pretrained('zhengpeng7/BiRefNet_lite')
85
  ```
86
 
87
  #### Use codes from GitHub + weights from HuggingFace