search

RUC-NLPIR / FlashRAG

⚡FlashRAG: A Python Toolkit for Efficient RAG Research
https://arxiv.org/abs/2405.13576
MIT License
891 stars 69 forks source link

Hello, I encountered the following bug while using the llama3-70B chat model to run the simple_pipeline.py script #13

Closed lwj2001 closed 1 month ago

lwj2001 commented 1 month ago

My search method is to set "bge". The bge model path is downloaded from the modelscope, and its model ID is bge-small-en-v1.5

Traceback (most recent call last):
  File "/Data1/home/fanziqi/func_eval/FlashRAG/examples/quick_start/simple_pipeline.py", line 35, in <module>
    output_dataset = pipeline.run(test_data,do_eval=True)
  File "/Data1/home/fanziqi/func_eval/FlashRAG/flashrag/pipeline/pipeline.py", line 80, in run
    retrieval_results = self.retriever.batch_search(input_query)
  File "/Data1/home/fanziqi/func_eval/FlashRAG/flashrag/retriever/retriever.py", line 60, in wrapper
    results, scores = func(self, query_list, num, True)
  File "/Data1/home/fanziqi/func_eval/FlashRAG/flashrag/retriever/retriever.py", line 90, in wrapper
    results, scores = func(self, query_list, num, True)
  File "/Data1/home/fanziqi/func_eval/FlashRAG/flashrag/retriever/retriever.py", line 158, in batch_search
    return self._batch_search(*args, **kwargs)
  File "/Data1/home/fanziqi/func_eval/FlashRAG/flashrag/retriever/retriever.py", line 279, in _batch_search
    batch_scores, batch_idxs = self.index.search(batch_emb, k=num)
  File "/Data1/home/fanziqi/.conda/envs/flashrag/lib/python3.9/site-packages/faiss/class_wrappers.py", line 329, in replacement_search
    assert d == self.d
AssertionError