Sentence Similarity
Safetensors
Japanese
RAGatouille
bert
ColBERT

Problem with readme

#1
by freQuensy23 - opened

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.)

Owner

Oops, thank your for flagging! This is a typo, the command should be search_encoded_docs. I'll go and update the README now!

bclavie changed discussion status to closed

Sign up or log in to comment