facebook
/

xm_transformer_s2ut_800m-es-en-st-asr-bt_h1_2022

speech-to-speech-translation

Model card Files Files and versions Community

Pengwei Li commited on Sep 20, 2022

Commit

ae4accc

•

1 Parent(s): eeb92fe

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -46,8 +46,8 @@ models, cfg, task = load_model_ensemble_and_task_from_hf_hub(
     arg_overrides={"config_yaml": "config.yaml", "task": "speech_to_text"},
     cache_dir=cache_dir,
 )
-model = models[0].cpu()
-cfg["task"].cpu = True
 generator = task.build_generator([model], cfg)

     arg_overrides={"config_yaml": "config.yaml", "task": "speech_to_text"},
     cache_dir=cache_dir,
 )
+#model = models[0].cpu()
+#cfg["task"].cpu = True
 generator = task.build_generator([model], cfg)