sanchit-gandhi HF staff commited on
Commit
05da548
1 Parent(s): 6fcef04

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -4
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. `model_name_or_card`: SeamlessM4T checkpoint. Can be either `multitask_unity_medium` for the medium model, or `multitask_unity_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,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("multitask_unity_large", 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.
 
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_large", 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.