def METHOD_NAME(self):
    """
    Construct the input spec for the convert dygraph model to static model.
    """
    self._input_spec = [paddle.static.InputSpec(shape=[None, None], dtype="int64", name="token_ids")]