Method __init__

fastdeploy/model_executor/layers/pooler.py:424–431

(
        self,
        model_config: ModelConfig,
    )

class StepPooler(Pooler):
    def __init__(
        self,
        model_config: ModelConfig,
    ) -> None:
        super().__init__()

        self.pooling = AllPool()
        self.head = RewardPoolerHead(model_config)

    def extract_states(
        self,