🔊 Confirm train hang

Signed-off-by: Evaline Ju <[email protected]>

caikit_nlp/modules/text_generation/text_generation_local.py

@@ -645,12 +645,12 @@ def _launch_training(
             train_dataset=training_dataset, **training_args
         )
 
+        # Confirm if stuck in training
+        assert False
+
         # Start training via Trainer.train function
         trainer.train()
 
-        # Check if training starts
-        assert False
-
         # save the model temporarily and reload it
         # this is done, since otherwise the model might be distributed in different
         # devices, in which case its better to use trainer's `prediction_step`