diff --git a/internlm/core/scheduler/no_pipeline_scheduler.py b/internlm/core/scheduler/no_pipeline_scheduler.py
index 9768790..6777acc 100644
--- a/internlm/core/scheduler/no_pipeline_scheduler.py
+++ b/internlm/core/scheduler/no_pipeline_scheduler.py
@@ -202,10 +202,8 @@ class NonPipelineScheduler(BaseScheduler):
             if return_output_label:
                 outputs.append(_output)
                 labels.append(_label)
-        
         if not return_output_label:
             outputs, labels = None, None
-        
         # Compatible for non-moe
         if hasattr(gpc.config.model, "num_experts"):
             return outputs, labels, loss, moe_loss