diff --git a/poetry.lock b/poetry.lock index 3334f98b..124fa464 100644 --- a/poetry.lock +++ b/poetry.lock @@ -1897,7 +1897,7 @@ testing = ["big-o", "jaraco.functools", "jaraco.itertools", "more-itertools", "p [metadata] lock-version = "1.1" python-versions = "^3.9" -content-hash = "d5d450ffed8640d77963042430262459f1111f183bb51c1e53f648cc383510a4" +content-hash = "ae4d3a67afa4aa799155cbd6d7a28d8e0192efc422a5307dca5fc6bc23cf041c" [metadata.files] absl-py = [ diff --git a/pyproject.toml b/pyproject.toml index 619ea462..e12a48ba 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -33,14 +33,13 @@ python = "^3.9" pytorch-lightning = "^2" torchmetrics = "^1" datasets = "^2.13" +transformers = "^4.18" +# we pin 2.0.0 because of https://github.com/pytorch/pytorch/issues/100974 +torch = ">=2.0.0, !=2.0.1" # pin to version below 2023.9.0 because that causes problems when using load_dataset with local files (e.g. json) fsspec = "<2023.9.0" -transformers = "^4" -# remove, for now -#huggingface-hub = "^0.5.1" +# required for testing, but, strangely, tests fail, if just in dev-dependencies absl-py = "^1.0.0" -# we pin 2.0.0 because of https://github.com/pytorch/pytorch/issues/100974 -torch = ">=2.0.0, !=2.0.1" [tool.poetry.dev-dependencies] Pygments = ">=2.10.0"