Skip to content

Add support for TerraTorch Geospatial models served via the vLLM /pooling endpoint#610

Draft
mgazz wants to merge 4 commits intovllm-project:mainfrom
mgazz:geo
Draft

Add support for TerraTorch Geospatial models served via the vLLM /pooling endpoint#610
mgazz wants to merge 4 commits intovllm-project:mainfrom
mgazz:geo

Conversation

@mgazz
Copy link

@mgazz mgazz commented Feb 25, 2026

Summary

This PR adds support for TerraTorch Geospatial models like Prithvi. Compared to traditional LLMs, Geospatial Models support image-to-image or tensor-to-tensor input/output, but they do not process text. Moreover they are served in vLLM at the /pooling endpoint.

Details

  • Support for the /pooling endpoint
  • A custom PoolingColumnMapper to map the prompt entry in datasets without creating conflict with the existing GenerativeColumnMapper
  • A new request handler for the pooling endpoint: PoolingRequestHandler

Test Plan

Related Issues


  • "I certify that all code in this PR is my own, except as noted below."

Use of AI

  • Includes AI-assisted code completion
  • Includes code generated by an AI application
  • Includes AI-generated tests (NOTE: AI written tests should have a docstring that includes ## WRITTEN BY AI ##)

…ling endpoint

Signed-off-by: Michele Gazzetti <michele.gazzetti1@ibm.com>
Signed-off-by: Michele Gazzetti <michele.gazzetti1@ibm.com>
Signed-off-by: Michele Gazzetti <michele.gazzetti1@ibm.com>
Signed-off-by: Michele Gazzetti <michele.gazzetti1@ibm.com>
@dbutenhof dbutenhof changed the title Add support for TerraTorch Geospatial models served via the vLLM /poling endpoint Add support for TerraTorch Geospatial models served via the vLLM /pooling endpoint Feb 27, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Add support for Geospatial models (served via vLLM)

1 participant