Commit e1da3ee

christian-pinto

authored and

committed

[Model] IBM/NASA Prithvi Geospatial model (vllm-project#12830)

Signed-off-by: SzymonOzog <szymon.ozog@aleph-alpha.com>

1 parent 60a08d9 commit e1da3eeCopy full SHA for e1da3ee

7 files changed

+811

-9

lines changed

examples/offline_inference
- prithvi_geospatial_mae.py
tests/models
- registry.py
vllm
- attention/backends
  - placeholder_attn.py
- inputs
  - preprocess.py
- model_executor/models
  - prithvi_geospatial_mae.py
  - registry.py
- worker
  - pooling_model_runner.py

7 files changed

+811

-9

lines changed

Comments

(0)