search

lightonai / pylate

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

update-colbert-inheritance #34

Closed raphaelsty closed 3 months ago

raphaelsty commented 3 months ago

Inherit from Sentence Transformer in order to reduce the amount of code unused in ColBERT models

NohTow commented 3 months ago

Fixed the loading of variables done in the init being overridden by the reading of the variables. Now it works great and the training seems ok, merging.