diff --git a/requirements.txt b/requirements.txt index 0639de6..fdfbde3 100644 --- a/requirements.txt +++ b/requirements.txt @@ -5,5 +5,5 @@ PyYAML>=6.0 dgl==2.1.0 pydantic>=1.10,<2 requests>=2.31 -transformers==4.37.2 +transformers==5.3.0 safetensors diff --git a/setup.cfg b/setup.cfg index cb5704b..a622ef5 100644 --- a/setup.cfg +++ b/setup.cfg @@ -21,7 +21,7 @@ install_requires = gnn = torch torch-geometric - transformers==4.37.2 + transformers==5.3.0 safetensors [options.packages.find]