diff --git a/pyproject.toml b/pyproject.toml index 8cbbea4..766ca0a 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -62,7 +62,7 @@ fasttext-wheel = {version = "*", optional = true} optuna = {version = "*", optional = true} matplotlib = {version = "*", optional = true} SoMaJo = {version = ">=2.3.1", optional = true} -torch = {version = "!=2.0.1", optional = true} # version 2.0.1 has poetry issues +torch = {version = "!=2.0.1,!=2.1.0", optional = true} # some versions have poetry issues transformers = {version = "*", optional = true} tiktoken = {version = "*", optional = true} safetensors = {version = "!=0.3.2", optional = true} # version 0.3.2 has poetry issues