diff --git a/requirements/rocm-test.txt b/requirements/rocm-test.txt
index 070c18363..af7703916 100644
--- a/requirements/rocm-test.txt
+++ b/requirements/rocm-test.txt
@@ -67,8 +67,6 @@ multiprocess==0.70.16
 # Required for v1/metrics/test_engine_logger_apis.py
 ray[cgraph,default]>=2.48.0
 
-# Plugins test
-terratorch @ git+https://github.com/IBM/terratorch.git@07184fcf91a1324f831ff521dd238d97fe350e3e
 torchgeo==0.7.0
     # via terratorch
 # MTEB Benchmark Test
@@ -98,3 +96,9 @@ transformers==4.57.3
 huggingface-hub==0.36.2
 # Pin Mistral Common
 mistral-common[image,audio]==1.9.1
+# Required for Prithvi tests
+terratorch==1.2.2
+# Required for Prithvi tests
+segmentation-models-pytorch==0.5.0
+# Required for Prithvi tests
+imagehash==4.3.2
diff --git a/vllm/model_executor/models/terratorch.py b/vllm/model_executor/models/terratorch.py
index 016cdd742..0dc778a09 100644
--- a/vllm/model_executor/models/terratorch.py
+++ b/vllm/model_executor/models/terratorch.py
@@ -209,7 +209,7 @@ class TerratorchMultiModalProcessor(BaseMultiModalProcessor[TerratorchProcessing
 
         _, passthrough_data = self._get_hf_mm_data(mm_items)
         mm_processed_data = BatchFeature(
-            {k: torch.tensor(v).unsqueeze(0) for k, v in passthrough_data.items()},
+            {k: torch.as_tensor(v).unsqueeze(0) for k, v in passthrough_data.items()},
             tensor_type="pt",
         )
         mm_placeholders = {"image": [PlaceholderRange(offset=0, length=0)]}