From 1d83b8dda0f5923fe0b668bb0bb0968da8d231bc Mon Sep 17 00:00:00 2001
From: tazlin <tazlin.on.github@gmail.com>
Date: Mon, 21 Oct 2024 11:50:09 -0400
Subject: [PATCH] fix: set torch 2.4.1 for rocm compat for now

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

diff --git a/requirements.txt b/requirements.txt
index 1b6dc6ac..591b9e92 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -4,7 +4,7 @@ horde_sdk>=0.15.0
 horde_model_reference>=0.9.1
 pydantic
 numpy==1.26.4
-torch>=2.5.0
+torch>=2.4.1
 # xformers>=0.0.19
 torchvision
 # torchaudio