From a380f06555489f6b94139e1df36a345aafca68a9 Mon Sep 17 00:00:00 2001
From: "l.gabrysiak" <l.gabrysiak@osadkowski.pl>
Date: Tue, 25 Feb 2025 12:32:03 +0100
Subject: [PATCH] modyfikacja trenera

---
 hft.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/hft.py b/hft.py
index b7ee50a..f960d70 100644
--- a/hft.py
+++ b/hft.py
@@ -130,7 +130,7 @@ training_args = TrainingArguments(
 trainer = CustomTrainer(
     model=model,
     args=training_args,
-    train_dataset=tokenized_dataset["train"],
+    train_dataset=tokenized_dataset,
 )
 
 # Trening modelu