Skip to content

0.2.27 TypeError: BertModel.forward() got an unexpected keyword argument 'token_lengths' #370

Description

@maxromanovsky

0.2.27 regresses BiEncoder.encode_labels: leaks token_lengths kwarg into BertModel.forward()

BiEncoderSpanGLiNER.inference(request.texts, request.labels, threshold=request.threshold) raises the following error:

TypeError: BertModel.forward() got an unexpected keyword argument 'token_lengths'

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions