diff --git a/speechbrain/inference/interfaces.py b/speechbrain/inference/interfaces.py index 005933bd84..3ee74981ac 100644 --- a/speechbrain/inference/interfaces.py +++ b/speechbrain/inference/interfaces.py @@ -141,7 +141,11 @@ def foreign_class( # For distributed setups, have this here: run_on_main( pretrainer.collect_files, - kwargs={"default_source": source, "use_auth_token": use_auth_token}, + kwargs={ + "default_source": source, + "use_auth_token": use_auth_token, + "local_strategy": local_strategy, + }, ) # Load on the CPU. Later the params can be moved elsewhere by specifying if not download_only: @@ -523,6 +527,7 @@ def from_hparams( kwargs={ "default_source": source, "use_auth_token": use_auth_token, + "local_strategy": local_strategy, }, ) # Load on the CPU. Later the params can be moved elsewhere by specifying