Restore, refactor, and improve the interpretability module

[tonywu71]

Description

Added

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

Tests

The following figures were obtained using the code from https://github.com/tonywu71/colpali-cookbooks/pull/5.

ColPali

ColQwen2

[tonywu71]

@ManuelFay I managed to make the interpretability work with both ColPali and ColQwen while keeping the API as simple as possible. Wdyt?

[ManuelFay]

@tonywu71 any reason this is not merged ?

[tonywu71]

@tonywu71 any reason this is not merged ?

Yup, was running a few extra tests yesterday. Gonna merge it tonight or tomorrow morning!