From 4d0dfc4996f0ad1440fcaac01c8185b7568e8c7a Mon Sep 17 00:00:00 2001 From: gkumbhat Date: Mon, 15 Jul 2024 15:43:49 -0500 Subject: [PATCH] :bug: Remove tee from torch launch config to correspond to API changeswith lib update Signed-off-by: gkumbhat --- caikit_nlp/toolkit/torch_run.py | 1 - 1 file changed, 1 deletion(-) diff --git a/caikit_nlp/toolkit/torch_run.py b/caikit_nlp/toolkit/torch_run.py index 43184f2d..d9725811 100644 --- a/caikit_nlp/toolkit/torch_run.py +++ b/caikit_nlp/toolkit/torch_run.py @@ -100,6 +100,5 @@ def get_torch_elastic_launch_config( rdzv_backend="static", rdzv_endpoint=f"{master_addr}:{master_port}", rdzv_configs=rdzv_configs, - tee=Std.ALL, max_restarts=max_restarts, )