lightonai / pylate

Late Interaction Models Training & Retrieval
https://lightonai.github.io/pylate/
MIT License
175 stars 7 forks source link

Creation of the PylateModelCard #67

Closed NohTow closed 1 week ago

NohTow commented 1 month ago

This PR fixes the issue with the model_card creation mentioned in #58 by creating a dedicated PylateModelCard. Besides fixing the issue, it also allows to create a card adapted to PyLate models while also having the original useful information for ST cards.

Let's first assert that everything is correct in the card before merging.

Edit: the failing test is due to the version of ST being too old. I'll rebase on main once we merge the PR for bumping the version.

raphaelsty commented 1 week ago

LGTM, hf_hub module will be useful :)

raphaelsty commented 1 week ago

LGTM

tomaarsen commented 1 week ago

Very exciting!