diff --git a/hft.py b/hft.py
index 86a3048..2ae6488 100644
--- a/hft.py
+++ b/hft.py
@@ -160,7 +160,7 @@ tokenized_dataset = dataset.map(tokenize_function, batched=True, batch_size=8)
 
 # Inicjalizacja modelu
 config = AutoModelForCausalLM.from_pretrained(model_name).config
-print("Vocabulary size:", config.vocab_size)
+#print("Vocabulary size:", config.vocab_size)
 model = CustomModel(model_name, config)
 model.to("cpu")  # Zmienione na CPU dla debugowania