Skip to content

Enable colqwen? #28

Description

@mbode

Would it be easily possible to enable using https://huggingface.co/vidore/colqwen2-v0.1 with byaldi?

Currently RAGMultiModalModel.from_pretrained("vidore/colqwen2-v0.1") leads to

ValueError('This pre-release version of Byaldi only supports ColPali for now. Incorrect model name specified.')

Activity

  1. feuler commented on Sep 27, 2024

    @feuler

    Would it be easily possible to enable using https://huggingface.co/vidore/colqwen2-v0.1 with byaldi?

    Currently RAGMultiModalModel.from_pretrained("vidore/colqwen2-v0.1") leads to

    ValueError('This pre-release version of Byaldi only supports ColPali for now. Incorrect model name specified.')

    I guess in the meanwhile you could change the file https://github.com/AnswerDotAI/byaldi/blob/main/byaldi/colpali.py
    to support it. (lines 10, 67, 81+82)
    See definitions: https://huggingface.co/vidore/colqwen2-v0.1
    Some things are different. e.g.:

    • from colpali_engine.models import ColQwen2, ColQwen2Processor
    • model = ColQwen2.from_pretrained...
    • processor = ColQwen2Processor.from_pretrained...
  2. jpetrantoni commented on Sep 28, 2024

    @jpetrantoni
    Contributor

    @feuler Thoughts on #29 ? Probably could check for correct model, init model, and init processor a little cleaner but didnt want to change too much.

  3. feuler commented on Sep 28, 2024

    @feuler

    @feuler Thoughts on #29 ? Probably could check for correct model, init model, and init processor a little cleaner but didnt want to change too much.

    Thank you very much ! Just tested it, and it seems to work fine.

  4. bclavie commented on Oct 3, 2024

    @bclavie
    Contributor

    Great contributions by @jpetrantoni who got it to work before I even recovered! Will ship a release very soon.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions