Apply suggestions from code review

Co-authored-by: Alex Brooks <alex.brooks@ibm.com>
Signed-off-by: Gaurav Kumbhat <kumbhat.gaurav@gmail.com>
Signed-off-by: gkumbhat <kumbhat.gaurav@gmail.com>

Author: gkumbhat

caikit_nlp/modules/text_generation/fine_tuning.py

@@ -100,7 +100,7 @@ def train(
             base_model:  Union[str, caikit_nlp.resources.pretrained_model.base.PretrainedModelBase]
                 Base resource model used for underlying generation.
             train_stream: DataStream[GenerationTrainRecord] or DataStream[ClassificationTrainRecord]
-                Data to be used for training the prompt vectors of the generation model.
+                Data to be used for fine-tuning the generation model.
             torch_dtype: str
                 TODO: Optional[Union[torch.dtype, str]]
                 Data type to use for training/inference of the underlying text generation model.

caikit_nlp/resources/pretrained_model/base.py

@@ -248,7 +248,7 @@ def get_trainer(
     ):
         """
         Args:
-            *kwargs: arguments supported by HF TrainingArguments:
+            **kwargs: arguments supported by HF TrainingArguments:
             https://huggingface.co/docs/transformers/v4.30.0/en/main_classes/trainer#transformers.TrainingArguments
 
         NOTE: following parameters are not supported currently: