From a060595a77c2cce242b66ccf9e68d11c93b2af43 Mon Sep 17 00:00:00 2001 From: PhilipMay Date: Fri, 13 Oct 2023 07:38:07 +0200 Subject: [PATCH] fix torch version issue --- pyproject.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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