fix: incorrect check on fast moe activation

Signed-off-by: Mehant Kammakomati <mehant.kammakomati2@ibm.com>

Author: kmehant

tuning/sft_trainer.py

@@ -167,8 +167,9 @@ def train(
                 "`--padding_free` argument was called with `packing=True`, "
                 "Trainer should not perform packing when using `--padding_free`"
             )
-
-    if fast_moe_config is not None:
+    if fast_moe_config is not None and fast_moe_config.fast_moe is None:
+        fast_moe_config = None
+    if fast_moe_config is not None and fast_moe_config.fast_moe is not None:
         # Checking for unsupported modules with Scatter MoE for LoRA
         # Only raise an error for `all-linear`
         restricted_modules = ["all-linear"]