bclavie

/

JaColBERTv2

like 16

I've tried to reproduce code from readme and get such error

from ragatouille import RAGPretrainedModel
RAG = RAGPretrainedModel.from_pretrained("bclavie/JaColBERTv2")

RAG.encode(['document_1', 'document_2'])
RAG.search_encoded_documents(query="your search query")

AttributeError Traceback (most recent call last)
Cell In[28], line 5
2 RAG = RAGPretrainedModel.from_pretrained("bclavie/JaColBERTv2")
4 RAG.encode(['document_1', 'document_2'])
----> 5 RAG.search_encoded_documents(query="your search query")

AttributeError: 'RAGPretrainedModel' object has no attribute 'search_encoded_documents'

ragatouille.version =='0.0.8post2'
Can you update readme and fix this bug in example code please.)