diff --git a/src/mBERT/training/model-training/requirements-minimal.txt b/src/mBERT/training/model-training/requirements-minimal.txt index cd2d843..16c2cdd 100644 --- a/src/mBERT/training/model-training/requirements-minimal.txt +++ b/src/mBERT/training/model-training/requirements-minimal.txt @@ -1,5 +1,5 @@ # Essential dependencies for mBERT training -torch>=2.0.0 +torch>=2.13.0 transformers>=4.50.0 scikit-learn>=1.5.0 pandas>=2.0.0 diff --git a/src/mBERT/training/model-training/requirements.txt b/src/mBERT/training/model-training/requirements.txt index 582665f..763b796 100644 --- a/src/mBERT/training/model-training/requirements.txt +++ b/src/mBERT/training/model-training/requirements.txt @@ -161,7 +161,7 @@ tokenizers==0.15.0 toml==0.10.2 tomli==2.0.1 tomlkit==0.11.6 -torch==2.9.1 +torch==2.13.0 tqdm==4.66.3 traitlets==5.9.0 transformers==4.53.0