alihan/MiniRAG
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix bugs

Browse Source
This commit is contained in:
tianyufan
2025-03-05 19:07:25 +08:00
parent 5d520fc5ba
commit 87c1dea5fe
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
reproduce/Step_0_index.py
View File

@@ -11,7 +11,7 @@ sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
from minirag import MiniRAG
from minirag.llm import (
gpt_4o_mini_complete,
hf_embedding,
hf_embed,
)
from minirag.utils import EmbeddingFunc
from transformers import AutoModel, AutoTokenizer
@@ -69,7 +69,7 @@ rag = MiniRAG(
embedding_func=EmbeddingFunc(
embedding_dim=384,
max_token_size=1000,
func=lambda texts: hf_embedding(
func=lambda texts: hf_embed(
texts,
tokenizer=AutoTokenizer.from_pretrained(EMBEDDING_MODEL),
embed_model=AutoModel.from_pretrained(EMBEDDING_MODEL),

4
reproduce/Step_1_QA.py
View File

@@ -13,7 +13,7 @@ from tqdm import trange
from minirag import MiniRAG, QueryParam
from minirag.llm import (
hf_model_complete,
hf_embedding,
hf_embed,
)
from minirag.utils import EmbeddingFunc
from transformers import AutoModel, AutoTokenizer
@@ -71,7 +71,7 @@ rag = MiniRAG(
embedding_func=EmbeddingFunc(
embedding_dim=384,
max_token_size=1000,
func=lambda texts: hf_embedding(
func=lambda texts: hf_embed(
texts,
tokenizer=AutoTokenizer.from_pretrained(EMBEDDING_MODEL),
embed_model=AutoModel.from_pretrained(EMBEDDING_MODEL),