🔊 More logs for text gen local train

Signed-off-by: Evaline Ju <[email protected]>
caikit · Feb 13, 2025 · ae2c373 · ae2c373
1 parent 763a663
commit ae2c373
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/caikit_nlp/modules/text_generation/text_generation_local.py b/caikit_nlp/modules/text_generation/text_generation_local.py
@@ -645,6 +645,8 @@ def _launch_training(
         trainer = base_model.get_trainer(
             train_dataset=training_dataset, **training_args
         )
+        print("Dir trainer!", dir(trainer))
+        print("Trainer attributes", trainer.__dict__)
 
         # Start training via Trainer.train function
         trainer.train()

diff --git a/tests/modules/text_generation/test_peft_prompt_tuning.py b/tests/modules/text_generation/test_peft_prompt_tuning.py
@@ -89,7 +89,6 @@ def test_run_stream_out_model(causal_lm_dummy_model):
     pred_stream = causal_lm_dummy_model.run_stream_out("This text doesn't matter")
     assert isinstance(pred_stream, Iterable)
     for pred in pred_stream:
-        print(pred)
         assert isinstance(pred, GeneratedTextStreamResult)