diff --git a/hft.py b/hft.py
index 7b8e6b7..f423943 100644
--- a/hft.py
+++ b/hft.py
@@ -2,7 +2,7 @@ import os
 import torch
 import torch.nn as nn
 #from transformers import AutoTokenizer, AutoModelForCausalLM, TrainingArguments, Trainer
-from transformers import GPTNeoForCausalLM, Trainer, TrainingArguments # Zmiana importu
+from transformers import GPTNeoForCausalLM, Trainer, TrainingArguments, AutoTokenizer # Zmiana importu
 from datasets import Dataset
 from PIL import Image
 import re