Przeglądaj źródła

Renaming multitask_* to seamlessM4T_*

Kaushik Ram Sadagopan 2 lat temu
rodzic
commit
0a08c9638d

+ 2 - 2
scripts/m4t/finetune/finetune.py

@@ -48,8 +48,8 @@ def init_parser() -> argparse.ArgumentParser:
     parser.add_argument(
         "--model_name",
         type=str,
-        default="multitask_unity_medium",
-        help="Base model name (`multitask_unity_medium`, `multitask_unity_large`)",
+        default="seamlessM4T_medium",
+        help="Base model name (`seamlessM4T_medium`, `seamlessM4T_large`)",
     )
     parser.add_argument(
         "--save_model_to",

+ 1 - 1
scripts/m4t/predict/README.md

@@ -41,7 +41,7 @@ python scripts/m4t/predict/predict.py <path_to_input_audio> asr <tgt_lang>
 
 ## Inference breakdown
 
-Inference calls for the `Translator` object instantiated with a Multitasking UnitY model with the options:
+Inference calls for the `Translator` object instantiated with a multitask UnitY model with the options:
 - `seamlessM4T_large`
 - `seamlessM4T_medium`
 

+ 1 - 1
scripts/m4t/predict/predict.py

@@ -36,7 +36,7 @@ def main():
         default=None,
     )
     parser.add_argument(
-        "--model_name", type=str, help="Model name", default="seamlessM4T_large"
+        "--model_name", type=str, help="Base model name (`seamlessM4T_medium`, `seamlessM4T_large`)", default="seamlessM4T_large"
     )
     parser.add_argument(
         "--vocoder_name", type=str, help="Vocoder name", default="vocoder_36langs"