Skip to content

allow user to swap encoders #12

@na50r

Description

@na50r

This code by default loads LaBSE when bertalign is imported. I think it would be more convennient to load the encoder when the align_sents method is called and give the user the option to specify a different model.

This could be convennient in cases where the user already knows with what languages they deal with and are running it on their CPU. Smaller encoders that support fewer languages and have smaller embedding dimensions run faster than LaBSE

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