Update check.py

check.py

@@ -5,7 +5,7 @@ from tqdm import tqdm
 
 # Завантаження моделей та токенізатора
 # original_model_name = "meta-llama/Meta-Llama-3.1-8B"
-original_model_name = "facebook/opt-350m"  # Це відкрита модель, яку можно використовувати для тестування
+original_model_name = "meta-llama/Meta-Llama-3.1-8B"
 
 fine_tuned_model_path = "./fine_tuned_model"  # Шлях до вашої донавченної моделі
 
@@ -17,7 +17,7 @@ fine_tuned_model = AutoModelForCausalLM.from_pretrained(fine_tuned_model_path)
 # test_dataset = load_dataset("wikitext", "wikitext-2-raw-v1", split="test")
 
 # Завантаження данних з локального тестового файлу
-with open("ilya_klimov_data.txt", "r", encoding="utf-8") as file:
+with open("m_and_m.txt", "r", encoding="utf-8") as file:
     text_data = file.read().strip()
 
 # Створення датасету