Commit 974dfd4

christian-pinto

authored

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

1 parent 3ee696a commit 974dfd4Copy full SHA for 974dfd4

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)