diff --git a/ch05/08_memory_efficient_weight_loading/memory-efficient-state-dict.ipynb b/ch05/08_memory_efficient_weight_loading/memory-efficient-state-dict.ipynb
index 9903393..9daf782 100644
--- a/ch05/08_memory_efficient_weight_loading/memory-efficient-state-dict.ipynb
+++ b/ch05/08_memory_efficient_weight_loading/memory-efficient-state-dict.ipynb
@@ -63,8 +63,7 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "memory_profiler version: 0.61.0\n",
-      "torch version: 2.4.1+cu121\n"
+      "torch version: 2.6.0\n"
      ]
     }
    ],