Skip to content

Commit 5d2361f

Browse files
committed
fix Prithvi
Signed-off-by: wang.yuqi <noooop@126.com>
1 parent 4fbd8bb commit 5d2361f

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

vllm/model_executor/models/prithvi_geospatial_mae.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,8 @@
2929
PoolerIdentity, SimplePooler)
3030
from vllm.model_executor.model_loader.weight_utils import default_weight_loader
3131
from vllm.model_executor.models.interfaces import (
32-
IsAttentionFree, MultiModalEmbeddings, SupportsMultiModalWithRawInput)
32+
IsAttentionFree, MultiModalEmbeddings, SupportsMultiModalWithRawInput,
33+
default_pooling_type)
3334
from vllm.model_executor.models.utils import AutoWeightsLoader
3435
from vllm.multimodal import MULTIMODAL_REGISTRY
3536
from vllm.multimodal.inputs import (MultiModalDataDict, MultiModalFieldConfig,
@@ -142,6 +143,7 @@ def apply(
142143
)
143144

144145

146+
@default_pooling_type("All")
145147
@MULTIMODAL_REGISTRY.register_processor(
146148
PrithviGeoSpatialMAEMultiModalProcessor,
147149
info=PrithviGeoSpatialMAEProcessingInfo,

0 commit comments

Comments
 (0)