chore(env): add torchinfo to environment.yml and requirements.txt

Author: neko941

.env/environment.yml

@@ -74,6 +74,7 @@ dependencies:
       - protobuf==7.34.1
       - psutil==7.2.2
       - ptflops==0.7.5
+      - torchinfo==1.8.0
       - Pygments==2.20.0
       - pyparsing==3.3.2
       - PySocks==1.7.1

.env/requirements.txt

@@ -43,3 +43,6 @@ timm~=1.0.26
 
 # TimeLLM (model)
 sentencepiece~=0.2.1
+
+# model summary
+torchinfo