Update my_model/fine_tuner/fine_tuner.py
Browse files
my_model/fine_tuner/fine_tuner.py
CHANGED
@@ -1,7 +1,7 @@
|
|
1 |
|
2 |
-
# Main Fine-Tuning Script for
|
3 |
|
4 |
-
# This script is the central executable for fine-tuning large language models, specifically designed for the
|
5 |
# model.
|
6 |
# It encompasses the entire process of fine-tuning, starting from data preparation to the final model training.
|
7 |
# The script leverages the 'FinetuningDataHandler' class for data loading, inspection, preparation, and splitting.
|
|
|
1 |
|
2 |
+
# Main Fine-Tuning Script for LLaMA-2-chat
|
3 |
|
4 |
+
# This script is the central executable for fine-tuning large language models, specifically designed for the LLaMA2 Chat
|
5 |
# model.
|
6 |
# It encompasses the entire process of fine-tuning, starting from data preparation to the final model training.
|
7 |
# The script leverages the 'FinetuningDataHandler' class for data loading, inspection, preparation, and splitting.
|