From ab671196134d5cf8ee006fdb44b46107613c178b Mon Sep 17 00:00:00 2001
From: Charlie Blake <charlieb@graphcore.ai>
Date: Mon, 15 Apr 2024 11:55:07 +0100
Subject: [PATCH] Fix torch version at 2.1 for now

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

diff --git a/requirements.txt b/requirements.txt
index 93bd878..da76b7e 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -4,4 +4,4 @@ einops
 numpy
 seaborn
 tabulate
-torch
\ No newline at end of file
+torch==2.1
\ No newline at end of file