Update finetunning.py

finetunning.py

@@ -1,6 +1,7 @@
 import os
+import time
 import torch
-from datasets import Dataset # , load_dataset
+from datasets import Dataset
 from transformers import (
     AutoTokenizer, 
     AutoModelForCausalLM, 
@@ -9,6 +10,8 @@ from transformers import (
     DataCollatorForLanguageModeling
 )
 
+start_time = time.time()
+
 # Створюємо директорії з правами на запис для всіх користувачів
 os.makedirs('./cache', exist_ok=True)
 os.chmod('./cache', 0o777)
@@ -65,5 +68,14 @@ trainer = Trainer(
 # Запуск fine-tuning
 trainer.train()
 
+print("LLM is finetunned");
+
 # Зберігання моделі
 trainer.save_model("./fine_tuned_model")
+
+end_time = time.time()
+
+training_time = end_time - start_time
+print(f"Донавчання зайняло {training_time:.2f} секунд")
+
+print("Донавчання пройшло вдало, модель сбережена у './fine_tuned_model'")