Merge branch 'ASUS-AICS:master' into print-estimate-tree-model-size-with-x-transpose

Author: ericliu8168

requirements_nn.txt

@@ -1,7 +1,8 @@
 nltk
 # wait for https://github.com/Lightning-AI/pytorch-lightning/pull/19191
 lightning==2.0.9
-torch
+# https://github.com/pytorch/text/releases
+torch<=2.3
 torchmetrics==0.10.3
 torchtext
 transformers

setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = libmultilabel
-version = 0.7.0
+version = 0.7.1
 author = LibMultiLabel Team
 license = MIT License
 license_file = LICENSE
@@ -39,7 +39,7 @@ python_requires = >=3.8
 nn =
     lightning==2.0.9
     nltk
-    torch
+    torch<=2.3
     torchmetrics==0.10.3
     torchtext
     transformers