Commit
•
3249b00
1
Parent(s):
cb513ed
Update README.md
Browse files
README.md
CHANGED
@@ -40,9 +40,9 @@ package can be installed by following the instructions outlined here: [Installat
|
|
40 |
|
41 |
Once installed, a [`Translator`](https://github.com/fairinternal/seamless_communication/blob/590547965b343b590d15847a0aa25a6779fc3753/src/seamless_communication/models/inference/translator.py#L47)
|
42 |
object can be instantiated to perform all five of the spoken langauge tasks. The `Translator` is instantiated with three arguments:
|
43 |
-
1.
|
44 |
-
2.
|
45 |
-
3.
|
46 |
|
47 |
```python
|
48 |
import torch
|
@@ -50,7 +50,7 @@ from seamless_communication.models.inference import Translator
|
|
50 |
|
51 |
|
52 |
# Initialize a Translator object with a multitask model, vocoder on the GPU.
|
53 |
-
translator = Translator("
|
54 |
```
|
55 |
|
56 |
Once instantiated, the `predict()` method can be used to run inference as many times on any of the supported tasks.
|
|
|
40 |
|
41 |
Once installed, a [`Translator`](https://github.com/fairinternal/seamless_communication/blob/590547965b343b590d15847a0aa25a6779fc3753/src/seamless_communication/models/inference/translator.py#L47)
|
42 |
object can be instantiated to perform all five of the spoken langauge tasks. The `Translator` is instantiated with three arguments:
|
43 |
+
1. **model_name_or_card**: SeamlessM4T checkpoint. Can be either `seamlessM4T_medium` for the medium model, or `seamlessM4T_large` for the large model
|
44 |
+
2. **vocoder_name_or_card**: vocoder checkpoint (`vocoder_36langs`)
|
45 |
+
3. **device**: Torch device
|
46 |
|
47 |
```python
|
48 |
import torch
|
|
|
50 |
|
51 |
|
52 |
# Initialize a Translator object with a multitask model, vocoder on the GPU.
|
53 |
+
translator = Translator("seamlessM4T_medium", vocoder_name_or_card="vocoder_36langs", device=torch.device("cuda:0"))
|
54 |
```
|
55 |
|
56 |
Once instantiated, the `predict()` method can be used to run inference as many times on any of the supported tasks.
|