def forward()

in src/sal/models/skywork_o1_prm/prm_model.py [0:0]

• 8 lines of code
• 1 McCabe index (conditional complexity)

    def forward(
        self,
        input_ids=None,
        past_key_values=None,
        attention_mask=None,
        return_past_key_values=False,
        return_probs=False,
        **kwargs,