Skip to content

v0.3.2: The interpretability update

Compare
Choose a tag to compare
@tonywu71 tonywu71 released this 17 Oct 08:57
· 23 commits to main since this release
347ab05

Description

✨ This release brings the interpretability module to colpali-engine and adds support for generating similarity maps with the ColQwen2 model.

🛠️ We’ve also made several code improvements and added tests for ColQwen2 to ensure better performance and reliability.

Features

Added

  • Restore, refactor, and improve interpretability module for generating similarity maps

Changed

  • Remove dummy image from ColPaliProcessor.process_queries

Fixed

  • Fix the compute_hardnegs.py script

Tests

  • Add missing model.eval() in tests
  • Add tests for ColQwen2

Full Changelog: v0.3.1...v0.3.2