Merge pull request #356 from Haidra-Org/main

fix: set torch 2.4.1 for rocm compat for now
Haidra-Org · Oct 21, 2024 · 69af49e · 69af49e
2 parents c710b5d + 1d83b8d
commit 69af49e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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