Skip to content

Conversation

@qnguyen3
Copy link
Contributor

@qnguyen3 qnguyen3 commented May 26, 2025

Hello @Blaizzy , please check out this PR for ColQwen2.5, which brings multimodal document retrieval to MLX Embeddings :)

However, I would recommend to wait for this PR to get merged first by HuggingFace to transformers so that we have a proper support for the AutoProcessor :)

Please let me know if there should be any additional. Looking forward to get this merge :D

@qnguyen3 qnguyen3 marked this pull request as draft May 26, 2025 22:51
@qnguyen3 qnguyen3 closed this May 26, 2025
@qnguyen3 qnguyen3 reopened this May 26, 2025
@qnguyen3 qnguyen3 marked this pull request as ready for review May 26, 2025 22:53
Copy link
Owner

@Blaizzy Blaizzy left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thank you very much Quan, this is really awesome!

I just have one tiny nit and after that we are ready to merge

- Removed the MODEL_REMAPPING dictionary as it was no longer needed.
- Updated the quantize_model function to accept skip_vision as a parameter.
- Refactored ColQwen2_5Config to ModelArgs for better compatibility with utils.py.
- Enhanced from_dict method to cleanly extract and convert text and vision configurations.
@qnguyen3
Copy link
Contributor Author

All done, everything should be fixed now @Blaizzy :)

@qnguyen3 qnguyen3 requested a review from Blaizzy May 27, 2025 21:46
@Blaizzy
Copy link
Owner

Blaizzy commented May 27, 2025

Thanks!

Are you able to load the quantized models?

If not it's because the predicate for quants has to be applied in the load_model function as well.

Copy link
Owner

@Blaizzy Blaizzy left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM!

@Blaizzy Blaizzy merged commit c8605d4 into Blaizzy:main May 27, 2025
1 check passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants