PeiqingYang commited on
Commit
0541cde
·
verified ·
1 Parent(s): 93745ed

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -4
README.md CHANGED
@@ -12,16 +12,16 @@ This repository contains the model described in [MatAnyone: Stable Video Matting
12
  Code: https://github.com/pq-yang/MatAnyone
13
 
14
  ## Usage
15
- `pip install huggingface_hub` required.
16
 
17
  ```shell
18
  pip install -q git+https://github.com/pq-yang/MatAnyone
19
  ```
20
 
21
- Users will only need to use these 2 lines to load and instantiate the model.
22
  ```python
23
  from matanyone.model.matanyone import MatAnyone
24
- model = MatAnyone.from_pretrained("PeiqingYang/MatAnyone")
25
  ```
26
 
27
- For inference, see the [inference script](https://github.com/pq-yang/MatAnyone/blob/main/inference_matanyone.py).
 
12
  Code: https://github.com/pq-yang/MatAnyone
13
 
14
  ## Usage
15
+ Requires `pip install huggingface_hub`.
16
 
17
  ```shell
18
  pip install -q git+https://github.com/pq-yang/MatAnyone
19
  ```
20
 
21
+ Users will only need to use these two lines to load and instantiate the model.
22
  ```python
23
  from matanyone.model.matanyone import MatAnyone
24
+ matanyone = MatAnyone.from_pretrained("PeiqingYang/MatAnyone").cuda().eval()
25
  ```
26
 
27
+ For inference, you may refer to the [inference script](https://github.com/pq-yang/MatAnyone/blob/main/inference_matanyone.py) and substitute L23-L28 (model loading part) with the above lines.