From 6d945675431556cefb3fd68730e50900931a3160 Mon Sep 17 00:00:00 2001
From: Yevhenii Semendiak <semendyak@gmail.com>
Date: Wed, 12 Jan 2022 18:31:02 +0200
Subject: [PATCH] Fix PyTorch version

---
 requirements.txt | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/requirements.txt b/requirements.txt
index 6cb79d3..7693b48 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,4 @@
 numpy>=1.13.3
-torch>=0.4.0
+torch>=0.4.0,<=1.4.0
 transformers>=3.5.1,<4.0.0
 sklearn