diff --git a/Fine_tune_a_Mistral_7b_model_with_DPO.ipynb b/Fine_tune_a_Mistral_7b_model_with_DPO.ipynb
index 788a43c..fa1f462 100644
--- a/Fine_tune_a_Mistral_7b_model_with_DPO.ipynb
+++ b/Fine_tune_a_Mistral_7b_model_with_DPO.ipynb
@@ -436,7 +436,7 @@
           "output_type": "stream",
           "name": "stderr",
           "text": [
-            "/usr/local/lib/python3.10/dist-packages/trl/trainer/ppo_config.py:141: UserWarning: The `optimize_cuda_cache` arguement will be deprecated soon, please use `optimize_device_cache` instead.\n",
+            "/usr/local/lib/python3.10/dist-packages/trl/trainer/ppo_config.py:141: UserWarning: The `optimize_cuda_cache` argument will be deprecated soon, please use `optimize_device_cache` instead.\n",
             "  warnings.warn(\n",
             "\u001b[34m\u001b[1mwandb\u001b[0m: Currently logged in as: \u001b[33mmlabonne\u001b[0m. Use \u001b[1m`wandb login --relogin`\u001b[0m to force relogin\n",
             "\u001b[34m\u001b[1mwandb\u001b[0m: \u001b[33mWARNING\u001b[0m If you're specifying your api key in code, ensure this code is not shared publicly.\n",
@@ -754,4 +754,4 @@
       ]
     }
   ]
-}
\ No newline at end of file
+}