From 75e1b877b9168f941c19e0507d533d3171c56b34 Mon Sep 17 00:00:00 2001 From: Ryan Rock Date: Fri, 21 Nov 2025 21:02:59 +0000 Subject: [PATCH 1/2] [CI/Build] Add terratorch for AMD Signed-off-by: Ryan Rock --- requirements/rocm-test.txt | 3 +++ 1 file changed, 3 insertions(+) diff --git a/requirements/rocm-test.txt b/requirements/rocm-test.txt index 432e11977872..7d798feaa676 100644 --- a/requirements/rocm-test.txt +++ b/requirements/rocm-test.txt @@ -39,3 +39,6 @@ mteb[bm25s]>=1.38.11, <2 # Required for eval tests lm-eval[api] @ git+https://github.com/EleutherAI/lm-evaluation-harness.git@206b7722158f58c35b7ffcd53b035fdbdda5126d + +# Plugins test +terratorch==1.1.1 From 0428b5a072e5e33affb1ad7935213ed5b697693a Mon Sep 17 00:00:00 2001 From: Ryan Rock Date: Fri, 21 Nov 2025 21:30:39 +0000 Subject: [PATCH 2/2] Apply Gemini suggestion; match cuda terratorch Signed-off-by: Ryan Rock --- requirements/rocm-test.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/requirements/rocm-test.txt b/requirements/rocm-test.txt index 7d798feaa676..eabb5065bfce 100644 --- a/requirements/rocm-test.txt +++ b/requirements/rocm-test.txt @@ -41,4 +41,4 @@ mteb[bm25s]>=1.38.11, <2 lm-eval[api] @ git+https://github.com/EleutherAI/lm-evaluation-harness.git@206b7722158f58c35b7ffcd53b035fdbdda5126d # Plugins test -terratorch==1.1.1 +terratorch @ git+https://github.com/IBM/terratorch.git@07184fcf91a1324f831ff521dd238d97fe350e3e