Unverified Commit 03a8770a authored by Andreas Karatzas's avatar Andreas Karatzas Committed by GitHub
Browse files

[ROCm][CI] Fix plugins test group; updating terratorch and dependencies (#34589)


Signed-off-by: default avatarAndreas Karatzas <akaratza@amd.com>
parent bc56a1d5
...@@ -67,8 +67,6 @@ multiprocess==0.70.16 ...@@ -67,8 +67,6 @@ multiprocess==0.70.16
# Required for v1/metrics/test_engine_logger_apis.py # Required for v1/metrics/test_engine_logger_apis.py
ray[cgraph,default]>=2.48.0 ray[cgraph,default]>=2.48.0
# Plugins test
terratorch @ git+https://github.com/IBM/terratorch.git@07184fcf91a1324f831ff521dd238d97fe350e3e
torchgeo==0.7.0 torchgeo==0.7.0
# via terratorch # via terratorch
# MTEB Benchmark Test # MTEB Benchmark Test
...@@ -98,3 +96,9 @@ transformers==4.57.3 ...@@ -98,3 +96,9 @@ transformers==4.57.3
huggingface-hub==0.36.2 huggingface-hub==0.36.2
# Pin Mistral Common # Pin Mistral Common
mistral-common[image,audio]==1.9.1 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
...@@ -209,7 +209,7 @@ class TerratorchMultiModalProcessor(BaseMultiModalProcessor[TerratorchProcessing ...@@ -209,7 +209,7 @@ class TerratorchMultiModalProcessor(BaseMultiModalProcessor[TerratorchProcessing
_, passthrough_data = self._get_hf_mm_data(mm_items) _, passthrough_data = self._get_hf_mm_data(mm_items)
mm_processed_data = BatchFeature( 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", tensor_type="pt",
) )
mm_placeholders = {"image": [PlaceholderRange(offset=0, length=0)]} mm_placeholders = {"image": [PlaceholderRange(offset=0, length=0)]}
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment