diff --git a/hft.py b/hft.py
index 2877c67..6c66b01 100644
--- a/hft.py
+++ b/hft.py
@@ -162,7 +162,7 @@ tokenized_dataset = dataset.map(tokenize_function, batched=True, batch_size=8)
 config = AutoModelForCausalLM.from_pretrained(model_name).config
 #print("Vocabulary size:", config.vocab_size)
 model = CustomModel(model_name, config)
-model.to("cpu")  # Zmienione na CPU dla debugowania
+#model.to("cpu")  # Zmienione na CPU dla debugowania
 
 # Konfiguracja treningu
 training_args = TrainingArguments(