diff --git a/hft.py b/hft.py
index 7f0d415..6e921fd 100644
--- a/hft.py
+++ b/hft.py
@@ -196,8 +196,7 @@ training_args = TrainingArguments(
     save_strategy="steps",
     save_steps=1000,
     logging_strategy="no",
-    report_to="none",
-    pin_memory=True,
+    report_to="none"
 )
 
 # Trening