Method __init__

fastdeploy/model_executor/layers/pooler.py:498–502

(self, pooling: PoolingMethod, head: PoolerHead)

        return cls(pooling, head)

    def __init__(self, pooling: PoolingMethod, head: PoolerHead) -> None:
        super().__init__()

        self.pooling = pooling
        self.head = head

    def get_supported_tasks(self) -> Set[PoolingTask]:
        return self.pooling.get_supported_tasks()