Update my_model/fine_tuner/fine_tuning_data_handler.py

my_model/fine_tuner/fine_tuning_data_handler.py

@@ -9,7 +9,7 @@ from my_model.LLAMA2.LLAMA2_model import Llama2ModelManager
 
 class FinetuningDataHandler:
     """
-    A class dedicated to handling data for fine-tuning language models. It manages loading,
+    A class dedicated to handling data for fine-tuning LLaMA-2 Chat models. It manages loading,
     inspecting, preparing, and splitting the dataset, specifically designed to filter out
     data samples exceeding a specified token count limit. This is crucial for models with
     token count constraints and it helps control the level of GPU RAM tolerance based on the number of tokens,