def _load_pooling_method(self) -> Callable:
        """Load the pooling method for the model."""
        if self.model_properties.pooling_method == PoolingMethod.Mean:
            return self._average_pool_func
        elif self.model_properties.pooling_method == PoolingMethod.CLS:
            return self._cls_pool_func
        else:
            raise InternalError(f"Invalid pooling method: {self.model_properties.pooling_method}")
Load the pooling method for the model.
src/marqo/inference/native_inference/embedding_models/hugging_face_model.py