From 419753096b6a5617e550a86aefb9c83f6c73e866 Mon Sep 17 00:00:00 2001
From: tazlin <tazlin.on.github@gmail.com>
Date: Mon, 30 Sep 2024 16:37:40 -0400
Subject: [PATCH] feat: use `torch>=2.4.1` and `cu124` by default

---
 requirements.txt | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index 53a4f54..e3101db 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,10 +1,10 @@
 # Add this in for tox, comment out for build
---extra-index-url https://download.pytorch.org/whl/cu121
+--extra-index-url https://download.pytorch.org/whl/cu124
 horde_sdk>=0.14.3
 horde_model_reference>=0.9.0
 pydantic
 numpy==1.26.4
-torch>=2.3.1
+torch>=2.4.1
 # xformers>=0.0.19
 torchvision
 # torchaudio