书生实战营llamaindex+Internlm2 RAG实践

llamaindex+Internlm2 RAG实践

个人对RAG的理解,并不对大语言模型进行微调,而是为其增加一个知识库,该知识库的的输入可以是word,pdf等格式文字,这些文字通过词嵌入向量模块后被转化为词向量储存,当用户进行提问时,问题也会被转化为词向量,问题向量会在知识向量库中寻找最相关,也就是距离最近的知识向量,然后将原始的用户问题和该知识共同作为提示,输入到大模型中,生成最终问答。

实践:1、配置基础环境,此处适用的是30%的A100,python版本3.10,安装Llamaindex、Pytorch、源词向量模型Sentence Transformer等相关依赖库。

2、下载InternLM2 1.8B,并询问xtuner是什么,得到答案如下图:

可以看出效果不好,模型并没有学习过xtuner的知识。

因此使用RAG对模型进行优化。先安装词向量依赖

pip install llama-index-embeddings-huggingface llama-index-embeddings-instructor

然后通过git clone https://github.com/InternLM/xtuner.git命令,获取知识库xtuner/README_zh-CN.md。

之后新建llamaindex_RAG.py并运行,这次得到的回答非常准确。

然后是在web上对llm进行询问,这个需要ssh连接,参考L0中的Linux课程,否则无法打开local网址。创建app.py并运行,得到下图:

然后提问:

  • 3
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
根据提供的引用内容,可以得知prompt+RAG的流程如下: 1. 首先,使用Retriever部分在知识库中检索出top-k个匹配的文档zi。 2. 然后,将query和k个文档拼接起来作为QA的prompt,送入seq2seq模型。 3. seq2seq模型生成回复y。 4. 如果需要进行Re-rank,可以使用LLM来rerank,给LLM写好prompt即可。 下面是一个简单的示例代码,演示如何使用prompt+RAG: ```python from transformers import RagTokenizer, RagRetriever, RagSequenceForGeneration # 初始化tokenizer、retriever和seq2seq模型 tokenizer = RagTokenizer.from_pretrained('facebook/rag-token-base') retriever = RagRetriever.from_pretrained('facebook/rag-token-base', index_name='exact', use_dummy_dataset=True) model = RagSequenceForGeneration.from_pretrained('facebook/rag-token-base') # 设置query和context query = "What is the capital of France?" context = "France is a country located in Western Europe. Paris, the capital city of France, is known for its romantic ambiance and iconic landmarks such as the Eiffel Tower." # 使用Retriever部分检索top-k个匹配的文档 retrieved_docs = retriever(query) # 将query和k个文档拼接起来作为QA的prompt input_dict = tokenizer.prepare_seq2seq_batch(query, retrieved_docs[:2], return_tensors='pt') generated = model.generate(input_ids=input_dict['input_ids'], attention_mask=input_dict['attention_mask']) # 输出生成的回复 generated_text = tokenizer.batch_decode(generated, skip_special_tokens=True)[0] print(generated_text) ```
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值