biondizzle/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

[Misc] Update tests and examples for Prithvi/Terratorch models (#34416)

Browse Source 
Signed-off-by: Christian Pinto <christian.pinto@ibm.com>
Co-authored-by: Cyrus Leung <tlleungac@connect.ust.hk>
This commit is contained in:
Christian Pinto
2026-02-14 07:03:51 +00:00
committed by GitHub
parent d1ea65d0a1
commit 342a7cda2d
9 changed files with 162 additions and 131 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/models/multimodal/pooling/test_prithvi_mae.py
View File

@@ -40,7 +40,7 @@ def _run_test(
vllm_model.llm.encode(prompt, pooling_task="plugin")
MODELS = ["mgazz/Prithvi-EO-2.0-300M-TL-Sen1Floods11"]
MODELS = ["ibm-nasa-geospatial/Prithvi-EO-2.0-300M-TL-Sen1Floods11"]
@pytest.mark.core_model