Tried to use a ColBERT model trained using Pylate since it was built on top of Sentence Transformers. The model loaded (SetFitModel.from_pretrained) but failed in trainer.train():
[/usr/local/lib/python3.10/dist-packages/sentence_transformers/losses/CosineSimilarityLoss.py](https://localhost:8080/#)---> 76 embeddings = [self.model(sentence_feature)["sentence_embedding"] for sentence_feature in sentence_features]KeyError: 'sentence_embedding'
Are those models fully incompatible or there is a fix?
Tried to use a ColBERT model trained using Pylate since it was built on top of Sentence Transformers. The model loaded (
SetFitModel.from_pretrained
) but failed intrainer.train()
:[/usr/local/lib/python3.10/dist-packages/sentence_transformers/losses/CosineSimilarityLoss.py](https://localhost:8080/#)
---> 76 embeddings = [self.model(sentence_feature)["sentence_embedding"] for sentence_feature in sentence_features]
KeyError: 'sentence_embedding'
Are those models fully incompatible or there is a fix?