def METHOD_NAME(self, ctxt_inputs: Dict[str, Any]) -> Dict[str, Any]:
    """
    Override PolyEncoderModule's inputs into the context encoder.
    """
    assert set(ctxt_inputs.keys()) == {'ctxt_tokens', 'ctxt_image'}
    return {
        'src_tokens': ctxt_inputs['ctxt_tokens'],
        'image_features': ctxt_inputs['ctxt_image'],
    }