diff --git a/fast_llm/engine/distributed/distributed.py b/fast_llm/engine/distributed/distributed.py index d93e17d1c..7b9d1e75d 100644 --- a/fast_llm/engine/distributed/distributed.py +++ b/fast_llm/engine/distributed/distributed.py @@ -165,7 +165,7 @@ def __init__(self, config: DistributedConfig, use_cpu: bool = False): self._config.local_world_size, self._config.timeout, use_cpu, - self._config.backend, + backend=self._config.backend, ) else: self._pool = _default_pool