From 1b4449f61960582e1b1cf47b27b16b64e15789db Mon Sep 17 00:00:00 2001
From: "l.gabrysiak" <l.gabrysiak@osadkowski.pl>
Date: Tue, 25 Feb 2025 22:03:59 +0100
Subject: [PATCH] mod

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

diff --git a/hft.py b/hft.py
index be2a268..5679524 100644
--- a/hft.py
+++ b/hft.py
@@ -245,7 +245,7 @@ def main():
         remove_unused_columns=False
     )
 
-    trainer = CustomTrainer(
+    trainer = Trainer(
         model=model,
         args=training_args,
         train_dataset=tokenized_dataset,