diff --git a/.devcontainer/requirements.txt b/.devcontainer/requirements.txt
index 448c5062..9c3e6bee 100644
--- a/.devcontainer/requirements.txt
+++ b/.devcontainer/requirements.txt
@@ -17,6 +17,6 @@ tensorflow-text==2.13.0
 tensorflow==2.13.1
 tensorboard==2.13.0
 tokenizers==0.19.1
-torchinfo==0.3.0
+torchinfo==1.8.0
 tqdm==4.66.5
 transformers==4.44.0