From 78d81789f849481bc740726a36b57ab977ff5bbb Mon Sep 17 00:00:00 2001 From: alexandremuzio Date: Mon, 22 Nov 2021 10:25:15 -0800 Subject: [PATCH] Removing `ImportError` from tutel import try/except --- deepspeed/moe/sharded_moe.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deepspeed/moe/sharded_moe.py b/deepspeed/moe/sharded_moe.py index 9b96555e30b2..9e8656f434c3 100644 --- a/deepspeed/moe/sharded_moe.py +++ b/deepspeed/moe/sharded_moe.py @@ -37,7 +37,7 @@ # python3 -m pip install --user --upgrade git+https://github.com/microsoft/tutel@v0.1.x from tutel import moe as tutel_moe TUTEL_INSTALLED = True -except ImportError: +except: # Fail silently so we don't spam logs unnecessarily if user isn't using tutel TUTEL_INSTALLED = False pass