diff --git a/CHANGELOG.md b/CHANGELOG.md index 8930d549a859f437a4cf3b5cf0b17b806b135c43..35f32d84d59c48a94cfa3a1c8dde4a9aa6e259e2 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -2,6 +2,10 @@ # PolDeepNer2 Changelog +## 0.8.2 +### Changed +- Updated verions of requirements + ## 0.8.1 ### Added - Pre-trained model pdn2-v08-nkjp-large diff --git a/requirements.txt b/requirements.txt index 9782ad65dc996fef800aa882fe4340445eda3922..18a6ce287f1ee1b4721d6187e3d124d5c7bb9f1d 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,10 +1,10 @@ +torch==2.0.0 numpy==1.19.4 -pytorch-transformers +pytorch-transformers==1.2.0 seqeval==0.0.12 tqdm fastapi==0.61.1 -PyYAML==5.3.1 +PyYAML==6.0 uvicorn==0.12.2 -pandas==1.1.1 -wandb==0.10.7 -transformers==4.16.2 \ No newline at end of file +wandb==0.10.7 +transformers==4.27.4 diff --git a/setup.py b/setup.py index 5c99b7e710594fc601dae33467ba85ef499df128..d60a9caa14ba43a8c712c2f92f80fa03c5172c96 100644 --- a/setup.py +++ b/setup.py @@ -4,15 +4,13 @@ with open("README.md", "r") as fh: long_description = fh.read() install_requires = [ - "numpy==1.19.4", + "torch==2.0.0", "pytorch-transformers==1.2.0", + "transformers==4.27.4", + "numpy==1.19.4", "seqeval==0.0.12", - "pytest~=6.0.1", - "flask==1.1.2", - "pandas==1.1.1", - "transformers==4.2.1", "tqdm", - "torch==1.12.1" + "pyyaml==6.0" ] setuptools.setup(