fix torch version issue

telekom · Oct 13, 2023 · a060595 · a060595
1 parent aea8a8f
commit a060595
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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