diff --git a/requirements_nn.txt b/requirements_nn.txt index 2da49767..f527d380 100644 --- a/requirements_nn.txt +++ b/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 diff --git a/setup.cfg b/setup.cfg index 927d596c..58c079a3 100644 --- a/setup.cfg +++ b/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