跳转至

11 省下钱买显卡,如何利用开源模型节约成本?

你好,我是徐文浩。

不知道课程上到这里,你账户里免费的5美元的额度还剩下多少了?如果你尝试着完成我给的几个数据集里的思考题,相信这个额度应该是不太够用的。而ChatCompletion的接口,又需要传入大量的上下文信息,实际消耗的Token数量其实比我们感觉的要多。

而且,除了费用之外,还有一个问题是数据安全。因为每个国家的数据监管要求不同,并不是所有的数据,都适合通过OpenAI的API来处理的。所以,从这两个角度出发,我们需要一个OpenAI以外的解决方案。那对于没有足够技术储备的中小型公司来说,最可行的一个思路就是利用好开源的大语言模型。

在Colab里使用GPU

因为这一讲我们要使用一些开源模型,但不是所有人的电脑里都有一个强劲的NVidia GPU的。所以,我建议你通过Colab来运行对应的Notebook,并且注意,要把对应的运行环境设置成GPU。

图片

  1. 你先选择菜单栏里的Runtime,然后点击Change runtime type。

图片

  1. 然后在弹出的对话框里,把Hardware accelerator换成GPU,然后点击Save就可以了。

只要用得不是太多,Colab的GPU是可以免费使用的。

HuggingfaceEmbedding,你的开源伙伴

其实我们之前在 第 4 讲对比零样本分类效果的时候,就已经使用过Google开源的模型T5了。那个模型的效果,虽然比OpenAI的API还是要差一些,但是其实90%的准确率也还算不错了。那么联想一下,上一讲我们使用的llama-index向量搜索部分,是不是可以用开源模型的Embedding给替换掉呢?

当然是可以的,llama-index支持你自己直接定义一个定制化的Embedding,对应的代码我放在了下面。

conda install -c conda-forge sentence-transformers

注:我们需要先安装一下sentence-transformers这个库。

import openai, os
import faiss
from llama_index import SimpleDirectoryReader, LangchainEmbedding, GPTFaissIndex, ServiceContext
from langchain.embeddings.huggingface import HuggingFaceEmbeddings
from langchain.text_splitter import CharacterTextSplitter
from llama_index.node_parser import SimpleNodeParser

openai.api_key = ""

text_splitter = CharacterTextSplitter(separator="\n\n", chunk_size=100, chunk_overlap=20)
parser = SimpleNodeParser(text_splitter=text_splitter)
documents = SimpleDirectoryReader('./data/faq/').load_data()
nodes = parser.get_nodes_from_documents(documents)

embed_model = LangchainEmbedding(HuggingFaceEmbeddings(
    model_name="sentence-transformers/paraphrase-multilingual-mpnet-base-v2"
))
service_context = ServiceContext.from_defaults(embed_model=embed_model)

dimension = 768
faiss_index = faiss.IndexFlatIP(dimension)
index = GPTFaissIndex(nodes=nodes,faiss_index=faiss_index, service_context=service_context)

输出结果:

INFO:sentence_transformers.SentenceTransformer:Load pretrained SentenceTransformer: sentence-transformers/paraphrase-multilingual-mpnet-base-v2
INFO:sentence_transformers.SentenceTransformer:Use pytorch device: cpu
WARNING:root:Created a chunk of size 130, which is longer than the specified 100
……
INFO:llama_index.token_counter.token_counter:> [build_index_from_documents] Total LLM token usage: 0 tokens
INFO:llama_index.token_counter.token_counter:> [build_index_from_documents] Total embedding token usage: 3198 tokens

在这个例子里面,我们使用了一个面向电商的FAQ的纯文本文件作为输入。里面是一系列预设好的FAQ问答对。为了确保我们没有使用OpenAI的API,我们先把openai.api_key给设成了一个空字符串。然后,我们定义了一个embeded_model,这个embeded_model里面,我们包装的是一个HuggingFaceEmbeddings的类。

因为HuggingFace为基于transformers的模型定义了一个标准,所以大部分模型你只需要传入一个模型名称,HuggingFacebEmbedding这个类就会下载模型、加载模型,并通过模型来计算你输入的文本的Embedding。使用HuggingFace的好处是,你可以通过一套代码使用所有的transfomers类型的模型。

sentence-transformers 是目前效果最好的语义搜索类的模型,它在BERT的基础上采用了对比学习的方式,来区分文本语义的相似度,它包括了一系列的预训练模型。我们在这里,选用的是 sentence-transformers下面的 paraphrase-multilingual-mpnet-base-v2 模型。顾名思义,这个是一个支持多语言(multilingual)并且能把语句和段落(paraphrase)变成向量的一个模型。因为我们给的示例都是中文,所以选取了这个模型。你可以根据你要解决的实际问题,来选取一个适合自己的模型。

我们还是使用Faiss这个库来作为我们的向量索引库,所以需要指定一下向量的维度,paraphrase-multilingual-mpnet-base-v2 这个模型的维度是768,所以我们就把维度定义成768维。

相应的对文档的切分,我们使用的是CharacterTextSplitter,并且在参数上我们做了一些调整。

首先,我们把“\n\n”这样两个连续的换行符作为一段段文本的分隔符,因为我们的FAQ数据里,每一个问答对都有一个空行隔开,正好是连续两个换行。

然后,我们把chunk_size设置得比较小,只有100。这是因为我们所使用的开源模型是个小模型,这样我们才能在单机加载起来。它能够支持的输入长度有限,只有128个Token,超出的部分会进行截断处理。如果我们不设置chunk_size,llama-index会自动合并多个chunk变成一个段落。

其次,我们还增加了一个小小的参数,叫做chunk_overlap。这个参数代表我们自动合并小的文本片段的时候,可以接受多大程度的重叠。它的默认值是200,超过了单段文档的chunk_size,所以我们这里要把它设小一点,不然程序会报错。

我们可以在对应的verbose日志里看到,这里的Embedding使用了3198个Token,不过这些Token都是我们通过sentence_transformers类型的开源模型计算的,不需要花钱。你的成本就节约下来了。

在创建完整个索引之后,我们就可以拿一些常见的电商类型的FAQ问题试一试。

问题1:

from llama_index import QueryMode

openai.api_key = os.environ.get("OPENAI_API_KEY")

response = index.query(
    "请问你们海南能发货吗?", 
    mode=QueryMode.EMBEDDING,
    verbose=True, 
)
print(response)

输出结果:

> Got node text: Q: 支持哪些省份配送?
A: 我们支持全国大部分省份的配送,包括北京、上海、天津、重庆、河北、山西、辽宁、吉林、黑龙江、江苏、浙江、安徽、福建、江西、山东、河南、湖北、湖南、广东、海南、四川、贵州、云南、陕西、甘肃、青海、台湾、内蒙古、广西、西藏、宁夏和新疆...

INFO:llama_index.token_counter.token_counter:> [query] Total LLM token usage: 341 tokens
INFO:llama_index.token_counter.token_counter:> [query] Total embedding token usage: 24 tokens

是的,我们支持海南省的配送。

问题2:

response = index.query(
    "你们用哪些快递公司送货?", 
    mode=QueryMode.EMBEDDING,
    verbose=True, 
)
print(response)

输出结果:

> Got node text: Q: 提供哪些快递公司的服务?
A: 我们与顺丰速运、圆通速递、申通快递、韵达快递、中通快递、百世快递等多家知名快递公司合作。...
INFO:llama_index.token_counter.token_counter:> [query] Total LLM token usage: 281 tokens
INFO:llama_index.token_counter.token_counter:> [query] Total embedding token usage: 27 tokens

我们与顺丰速运、圆通速递、申通快递、韵达快递、中通快递、百世快递等多家知名快递公司合作,用他们的服务送货。

问题3:

response = index.query(
    "你们的退货政策是怎么样的?", 
    mode=QueryMode.EMBEDDING,
    verbose=True, 
)
print(response)

输出结果:

> Got node text: Q: 退货政策是什么?
A: 自收到商品之日起7天内,如产品未使用、包装完好,您可以申请退货。某些特殊商品可能不支持退货,请在购买前查看商品详情页面的退货政策。...
INFO:llama_index.token_counter.token_counter:> [query] Total LLM token usage: 393 tokens
INFO:llama_index.token_counter.token_counter:> [query] Total embedding token usage: 27 tokens

我们的退货政策是自收到商品之日起7天内,如产品未使用、包装完好,您可以申请退货。某些特殊商品可能不支持退货,请在购买前查看商品详情页面的退货政策。

我们在问问题的时候,指定了query的mode是Embedding。通过三个常用的问题,我们可以看到,AI都给出了正确的回答,效果还是不错的。

使用ChatGLM提供对话效果

通过上面的代码,我们已经把生成Embedding以及利用Embedding的相似度进行搜索搞定了。但是,我们在实际问答的过程中,使用的还是OpenAI的Completion API。那么这一部分我们有没有办法也替换掉呢?

同样的,我们寻求开源模型的帮助。在这里,我们就不妨来试一下来自清华大学的ChatGLM语言模型,看看中文的开源语言模型,是不是也有基本的知识理解和推理能力。

首先我们还是要安装一些依赖包,因为icetk我没有找到Conda的源,所以我们这里通过pip来安装,但是在Conda的包管理器里一样能够看到。

pip install icetk
pip install cpm_kernels

然后,我们还是先通过transformers来加载模型。ChatGLM 最大的一个模型有1300亿个参数。

from transformers import AutoTokenizer, AutoModel
tokenizer = AutoTokenizer.from_pretrained("THUDM/chatglm-6b-int4", trust_remote_code=True)
model = AutoModel.from_pretrained("THUDM/chatglm-6b-int4", trust_remote_code=True).half().cuda()
model = model.eval()

输出结果:

Explicitly passing a `revision` is encouraged when loading a model with custom code to ensure no malicious code has been contributed in a newer revision.
Explicitly passing a `revision` is encouraged when loading a configuration with custom code to ensure no malicious code has been contributed in a newer revision.
Explicitly passing a `revision` is encouraged when loading a model with custom code to ensure no malicious code has been contributed in a newer revision.
No compiled kernel found.
Compiling kernels : /root/.cache/huggingface/modules/transformers_modules/THUDM/chatglm-6b-int4/dac03c3ac833dab2845a569a9b7f6ac4e8c5dc9b/quantization_kernels.c
Compiling gcc -O3 -fPIC -std=c99 /root/.cache/huggingface/modules/transformers_modules/THUDM/chatglm-6b-int4/dac03c3ac833dab2845a569a9b7f6ac4e8c5dc9b/quantization_kernels.c -shared -o /root/.cache/huggingface/modules/transformers_modules/THUDM/chatglm-6b-int4/dac03c3ac833dab2845a569a9b7f6ac4e8c5dc9b/quantization_kernels.so
Kernels compiled : /root/.cache/huggingface/modules/transformers_modules/THUDM/chatglm-6b-int4/dac03c3ac833dab2845a569a9b7f6ac4e8c5dc9b/quantization_kernels.so
Load kernel : /root/.cache/huggingface/modules/transformers_modules/THUDM/chatglm-6b-int4/dac03c3ac833dab2845a569a9b7f6ac4e8c5dc9b/quantization_kernels.so
Using quantization cache
Applying quantization to glm layers

但是这么大的模型,无论是你自己的电脑,还是Colab提供的GPU和TPU显然都放不了。所以我们只能选用一个裁剪后的60亿个参数的版本,并且我们还必须用int-4量化的方式,而不是用float16的浮点数。所以,这里我们的模型名字就叫做 chatglm-6b-int4,也就是 6B的参数量,通过int-4量化。然后,在这里,我们希望通过GPU进行模型的计算,所以加载模型的时候调用了.cuda()。

这里加载模型的时候,我们还设置了一个 trust_remote_code = true 的参数,这是因为ChatGLM的模型不是一个Huggingface官方发布的模型,而是由用户贡献的,所以需要你显式确认你信任这个模型的代码,它不会造成恶意的破坏。我们反正是在Colab里面运行这个代码,所以倒是不用太担心。

如果你想要用CPU运行,可以把模型加载的代码换成下面这样。

model = AutoModel.from_pretrained("THUDM/chatglm-6b-int4",trust_remote_code=True).float()

不过,我不建议你这么做。你没有GPU的话,还是直接使用Colab的GPU就好了。因为CPU在运行对话的时候非常慢。

在拿到模型之后我们就可以尝试着通过这个模型来进行问答了。

问题1:

question = """
自收到商品之日起7天内,如产品未使用、包装完好,您可以申请退货。某些特殊商品可能不支持退货,请在购买前查看商品详情页面的退货政策。

根据以上信息,请回答下面的问题:

Q: 你们的退货政策是怎么样的?
"""
response, history = model.chat(tokenizer, question, history=[])
print(response)

输出结果:

我们的退货政策是在产品未使用包装完好的情况下自收到商品之日起7天内可以退货请注意某些特殊商品可能不支持退货在购买前请查看商品详情页面的退货政策以了解具体情况

可以看到,ChatGLM的回答,的确是按照我们提示语里面有的上下文给出的。我们可以试着去掉提供的FAQ的内容,看看它的回答是怎样的。

问题2:

question = """
Q: 你们的退货政策是怎么样的?
A: 
"""
response, history = model.chat(tokenizer, question, history=[])
print(response)

输出结果:

我们公司的政策是如果您在购买后的7天内发现我们的产品不符合您的期望可以申请退货退货条件如下
1. 符合退货政策要求的产品
2. 未使用或损坏的产品
3. 符合退货地点的要求即退货地址与购买地址相同
4. 符合退货流程的要求即您需要提供有效的退货单并完成退货流程
如果符合上述退货政策要求我们将尽快处理您的退货申请并为您退款请注意如果您已经使用了我们的产品或者产品有损坏或变质等情况我们将无法为您提供退货服务

可以看到,在没有合适的上下文信息的时候,它编造出了一些原本没有的要求,比如“退货地址与购买地址相同”。

我们再来看一个例子,看看它能不能拥有简单的推理能力。我们的上下文里只说了可以送到海南,没有说是否支持三亚这个城市,看看这个时候AI能不能回答对这个问题。

问题3:

question = """
我们支持全国大部分省份的配送,包括北京、上海、天津、重庆、河北、山西、辽宁、吉林、黑龙江、江苏、浙江、安徽、福建、江西、山东、河南、湖北、湖南、广东、海南、四川、贵州、云南、陕西、甘肃、青海、台湾、内蒙古、广西、西藏、宁夏和新疆.

根据以上信息,请回答下面的问题:

Q: 你们能配送到三亚吗?
"""
response, history = model.chat(tokenizer, question, history=[])
print(response)

输出结果:

是的我们支持全国大部分省份的配送包括三亚市

可以看到,ChatGLM知道是可以配送到三亚的。不过万一是巧合呢?我们再看看在上下文里面,去掉了东三省,然后问问它能不能送到哈尔滨。

问题4:

question = """
我们支持全国大部分省份的配送,包括北京、上海、天津、重庆、河北、山西、江苏、浙江、安徽、福建、江西、山东、河南、湖北、湖南、广东、海南、四川、贵州、云南、陕西、甘肃、青海、台湾、内蒙古、广西、西藏、宁夏和新疆.但是不能配送到东三省

根据以上信息,请回答下面的问题:

Q: 你们能配送到哈尔滨吗?
"""
response, history = model.chat(tokenizer, question, history=[])
print(response)

回答:

很抱歉我们目前不能配送到哈尔滨

结果也是正确的,这个时候,ChatGLM会回答我们是送不到哈尔滨的。既然ChatGLM能够正确回答这个问题,那我们的FAQ问答就可以用ChatGLM来搞定了。

将ChatGLM封装成LLM

不过上面的代码里面,我们用的还是原始的ChatGLM的模型代码,还不能直接通过query来访问llama-index直接得到答案。要做到这一点倒也不难,我们把它封装成一个LLM类,让我们的index使用这个指定的大语言模型就好了。对应的 llama-index 的文档,你也可以自己去看一下。

import openai, os
import faiss
from llama_index import SimpleDirectoryReader, LangchainEmbedding, GPTFaissIndex, ServiceContext
from langchain.embeddings.huggingface import HuggingFaceEmbeddings
from langchain.text_splitter import CharacterTextSplitter
from llama_index.node_parser import SimpleNodeParser

from langchain.llms.base import LLM
from llama_index import LLMPredictor
from typing import Optional, List, Mapping, Any

class CustomLLM(LLM):
    def _call(self, prompt: str, stop: Optional[List[str]] = None) -> str:
        response, history = model.chat(tokenizer, prompt, history=[])
        return response

    @property
    def _identifying_params(self) -> Mapping[str, Any]:
        return {"name_of_model": "chatglm-6b-int4"}

    @property
    def _llm_type(self) -> str:
        return "custom"

我们把这个CustomLLM对象,传入index的构造函数里,重新运行一下我们的问题,看看效果是怎样的。

from langchain.text_splitter import SpacyTextSplitter

llm_predictor = LLMPredictor(llm=CustomLLM())

text_splitter = CharacterTextSplitter(separator="\n\n", chunk_size=100, chunk_overlap=20)
parser = SimpleNodeParser(text_splitter=text_splitter)
documents = SimpleDirectoryReader('./drive/MyDrive/colab_data/faq/').load_data()
nodes = parser.get_nodes_from_documents(documents)

embed_model = LangchainEmbedding(HuggingFaceEmbeddings(
    model_name="sentence-transformers/paraphrase-multilingual-mpnet-base-v2"
))
service_context = ServiceContext.from_defaults(embed_model=embed_model, llm_predictor=llm_predictor)

dimension = 768
faiss_index = faiss.IndexFlatIP(dimension)
index = GPTFaissIndex(nodes=nodes, faiss_index=faiss_index, service_context=service_context)
from llama_index import QuestionAnswerPrompt
from llama_index import QueryMode

QA_PROMPT_TMPL = (
    "{context_str}"
    "\n\n"
    "根据以上信息,请回答下面的问题:\n"
    "Q: {query_str}\n"
    )
QA_PROMPT = QuestionAnswerPrompt(QA_PROMPT_TMPL)

response = index.query(
    "请问你们海南能发货吗?", 
    mode=QueryMode.EMBEDDING,
    text_qa_template=QA_PROMPT,
    verbose=True, 
)
print(response)

输出结果:

> Got node text: Q: 支持哪些省份配送
A: 我们支持全国大部分省份的配送包括北京上海天津重庆河北山西辽宁吉林黑龙江江苏浙江安徽福建江西山东河南湖北湖南广东海南四川贵州云南陕西甘肃青海台湾内蒙古广西西藏宁夏和新疆...

海南能发货

可以看到,这样处理之后,我们就可以直接使用ChatGLM的模型,来进行我们的FAQ的问答了。

现在,我们有了一个通过paraphrase-multilingual-mpnet-base-v2模型来计算Embeddding并进行语义搜索,然后通过chatglm-6b-int4的模型来进行问答的解决方案了。而且这两个模型,可以跑在一块家用级别的显卡上。是不是很厉害?

开源模型的不足之处

看起来,我们这个本机就能运行的小模型似乎已经完成了。数据安全,又不用担心花费。但显然,事情没有那么简单。因为刚才我们处理的电商FAQ问题比较简单,我们再拿一个稍微复杂一点的问题来看看效果。

text_splitter = SpacyTextSplitter(pipeline="zh_core_web_sm", chunk_size = 128, chunk_overlap=32)
parser = SimpleNodeParser(text_splitter=text_splitter)
documents = SimpleDirectoryReader('./drive/MyDrive/colab_data/zhaohuaxishi/').load_data()
nodes = parser.get_nodes_from_documents(documents)

embed_model = LangchainEmbedding(HuggingFaceEmbeddings(
    model_name="sentence-transformers/paraphrase-multilingual-mpnet-base-v2"
))
service_context = ServiceContext.from_defaults(embed_model=embed_model, llm_predictor=llm_predictor)

dimension = 768
faiss_index = faiss.IndexFlatIP(dimension)
index = GPTFaissIndex(nodes=nodes, faiss_index=faiss_index, service_context=service_context)

输出结果:

INFO:sentence_transformers.SentenceTransformer:Load pretrained SentenceTransformer: sentence-transformers/paraphrase-multilingual-mpnet-base-v2
INFO:sentence_transformers.SentenceTransformer:Use pytorch device: cpu
……
INFO:llama_index.token_counter.token_counter:> [build_index_from_documents] Total LLM token usage: 0 tokens
INFO:llama_index.token_counter.token_counter:> [build_index_from_documents] Total embedding token usage: 91882 tokens

这一次,我们输入索引起来的数据,是鲁迅先生整套《朝花夕拾》的散文集。选用这个是因为对应作品的版权已经过了保护期。我们来看看,在这套文集的内容里面,使用我们上面的纯开源方案,效果会是怎样的。

对应的模型和索引加载的代码基本一致,只有一个小小的区别,就是在文本分割的时候,我们用了上一讲介绍过的SpacyTextSplitter,因为这里都是散文的内容,而不是确定好格式的QA对。所以通过SpacyTextSplitter来分句,并在允许的时候合并小的片段是有意义的。

然后,我们试着问一下上一讲我们问过的问题,看看效果怎么样。

问题1:

# query will use the same embed_model
from llama_index import QueryMode
from llama_index import QuestionAnswerPrompt

openai.api_key = os.environ.get("OPENAI_API_KEY")

QA_PROMPT_TMPL = (
    "下面的内容来自鲁迅先生的散文集《朝花夕拾》,很多内容是以第一人称写的 \n"
    "---------------------\n"
    "{context_str}"
    "\n---------------------\n"
    "根据这些信息,请回答问题: {query_str}\n"
    "如果您不知道的话,请回答不知道\n"
)
QA_PROMPT = QuestionAnswerPrompt(QA_PROMPT_TMPL)

response = index.query(
    "鲁迅先生在日本学习医学的老师是谁?", 
    mode=QueryMode.EMBEDDING,
    similarity_top_k = 1,
    text_qa_template=QA_PROMPT,
    verbose=True, 
)
print(response)

输出结果:

> Got node text: 一将书放在讲台上便用了缓慢而很有顿挫的声调向学生介绍自己道:—— 
    我就是叫作藤野严九郎的……。”


后面有几个人笑起来了
他接着便讲述解剖学在日本发达的历史那些大大小小的书便是从最初到现今关于这一门学问的著作...

鲁迅先生在日本学习医学的老师是藤野严九郎

问题2:

response = index.query(
    "鲁迅先生是在日本的哪个城市学习医学的?", 
    mode=QueryMode.EMBEDDING, 
    similarity_top_k = 1,   
    text_qa_template=QA_PROMPT,
    verbose=True, 
)
print(response)

输出结果:

> Got node text: 有时我常常想他的对于我的热心的希望不倦的教诲小而言之是为中国就是希望中国有新的医学大而言之是为学术就是希望新的医学传到中国去...

根据这些信息无法得出鲁迅先生是在日本的哪个城市学习医学的答案

可以看到,有些问题在这个模式下,定位到的文本片段是正确的。但是有些问题,虽然定位的还算是一个相关的片段,但是的确无法得出答案。

在这个过程中,我们可以观察到这样一个问题: 那就是单机的开源小模型能够承载的文本输入的长度问题。在我们使用OpenAI的gpt-3.5-turbo模型的时候,我们最长支持4096个Token,也就是一个文本片段可以放上上千字在里面。但是我们这里单机用的paraphrase-multilingual-mpnet-base-v2模型,只能支持128个Token的输入,虽然对应的Tokenizer不一样,但是就算一个字一个Token,也就100个字而已。这使得我们检索出来的内容的上下文太少了,很多时候没有足够的信息,让语言模型去回答。

当然,这个问题并不是无法弥补的。我们可以通过把更大规模的模型,部署到云端来解决。这个内容,我们课程的第三部分专门有一讲会讲解。

不过,有一个更难解决的问题,就是模型的推理能力问题。比如,我们可以再试试第 1 讲里给商品总结英文名称和卖点的例子。

question = """Consideration proudct : 工厂现货PVC充气青蛙夜市地摊热卖充气玩具发光蛙儿童水上玩具

1. Compose human readale product title used on Amazon in english within 20 words.
2. Write 5 selling points for the products in Amazon.
3. Evaluate a price range for this product in U.S.

Output the result in json format with three properties called title, selling_points and price_range"""
response, history = model.chat(tokenizer, question, history=[])
print(response)

输出结果:

1. title: 充气玩具青蛙夜市地摊卖
2. selling_points:
    - 工厂现货保证产品质量
    - PVC充气环保耐用
    - 夜市地摊方便销售
    - 热卖最受欢迎产品
    - 儿童水上玩具适合各种年龄段儿童
3. price_range: (in USD)
    - low:   $1.99
    - high:   $5.99

可以看到,虽然这个结果不算太离谱,多少和问题还是有些关系的。但是无论是翻译成英文,还是使用JSON返回,模型都没有做到。给到的卖点也没有任何“推理出来”的性质,都是简单地对标题的重复描述。即使你部署一个更大版本的模型到云端,也好不到哪里去。

这也是ChatGPT让人震撼的原因,的确目前它的效果还是要远远超出任何一个竞争对手和开源项目的。

小结

好了,最后我们来回顾一下。这一讲里,我们一起尝试用开源模型来代替ChatGPT。我们通过sentence_transfomers类型的模型,生成了文本分片的Embedding,并且基于这个Embedding来进行语义检索。我们通过 ChatGLM 这个开源模型,实现了基于上下文提示语的问答。在简单的电商QA这样的场景里,效果也还是不错的。即使我们使用的都是单机小模型,它也能正确回答出来。这些方法,也能节约我们的成本。不用把钱都交给OpenAI,可以攒着买显卡来训练自己的模型。

但是,当我们需要解决更加复杂的问题时,比如需要更长的上下文信息,或者需要模型本身更强的推理能力的时候,这样的小模型就远远不够用了。更长的上下文信息检索,我们还能够通过在云端部署更大规模的模型,解决部分问题。但是模型的推理能力,目前的确没有好的解决方案。

所以不得不佩服,OpenAI的在AGI这个目标上耕耘多年后震惊世人的效果。

思考题

最后,给你留一个思考题。ChatGLM并不是唯一的中文大语言模型,开源社区目前在快速推进,尝试用各种方式提供更好的开源大模型。比如基于斯坦福的Alpaca数据集进行微调的 Chinese-LLaMA-Alpaca,链家科技开源的 BELLE。你可以挑选一个模型试一试,看看它们的效果和ChatGLM比起来怎么样。欢迎你把你的评测结果分享出来,也欢迎你把这节课分享给需要的朋友,共同参谋,一起进步。我们下节课再见。

推荐阅读

基于开源模型来解决问题的思路并非我的原创,网上也有不少其他朋友用类似的方式解决了自己的问题。比如《让 LLM 回答问题更靠谱》这篇文章就组合了三个模型来完成了医学领域的语义搜索、语义匹配排序,以及最终的问答语句生成。你可以读一下。

精选留言(15)
  • Toni 👍(6) 💬(1)

    在第二讲中,用亚马逊提供的用户对一些食物评价的真实数据集进行了情感分析。当时为了避免重新调用 OpenAI 的 API 浪费钱,老师推荐使用已经计算好的含有 Embedding 的数据。用openai.embeddings_utils 中的 get_embedding (EMBEDDING_MODEL = "text-embedding-ada-002")不仅费钱还耗时。我试着跑过100个数据,好像用了20分钟,花费也不少。 本节课中老师介绍了免费的 sentence_transformers 正好可以拿来用在情感数据分析上, 选用 model = SentenceTransformer('paraphrase-multilingual-mpnet-base-v2')。同样计算1000个数据的 embedding,速度很快,且无费用,适合练手。 测试结果如下: precision recall f1-score support negative 0.77 0.78 0.77 148 positive 0.96 0.95 0.96 773 accuracy 0.93 921 macro avg 0.86 0.87 0.86 921 weighted avg 0.93 0.93 0.93 921 为了对比,将第二讲中老师用 OpenAI 的方法得到的结果放在了这里: precision recall f1-score support negative 0.98 0.73 0.84 136 positive 0.96 1.00 0.98 789 accuracy 0.96 925 macro avg 0.97 0.86 0.91 925 weighted avg 0.96 0.96 0.96 925 使用的方法不用,结果也不同。OpenAI 的准确率更高。

    2023-04-11

  • zhihai.tu 👍(5) 💬(1)

    老师,想请教一下,我发现我用colab跑了程序,如果第二天再打开这个程序,相应的已经跑过得脚本,效果都失效了,比如pip安装过的都还原了,另外磁盘目录上生成的文件夹和文件也没了。想请问下如何解决这个问题呢?不然相当的麻烦。谢谢。

    2023-04-26

  • 东方奇骥 👍(4) 💬(3)

    老师,是不是要4080显卡才跑得动?单机能跑得动吗

    2023-04-07

  • 胖子 👍(2) 💬(5)

    AttributeError Traceback (most recent call last) <ipython-input-54-086bfff09511> in <cell line: 8>() 6 Q: 你们的退货政策是怎么样的? 7 """ ----> 8 response, history = model.chat(tokenizer, question, history=[]) 9 print(response) 21 frames /usr/local/lib/python3.9/dist-packages/google/protobuf/unknown_fields.py in <module> 42 from google.protobuf.internal import api_implementation 43 ---> 44 if api_implementation._c_module is not None: # pylint: disable=protected-access 45 UnknownFieldSet = api_implementation._c_module.UnknownFieldSet # pylint: disable=protected-access 46 else: AttributeError: module 'google.protobuf.internal.api_implementation' has no attribute '_c_module' 搞不定

    2023-04-22

  • 川月 👍(2) 💬(2)

    使用这个开源模型获取的index怎么保存啊,使用之前的方法并不行,还有生成的index可以追加吗,不然每次有新数据的时候都要重新跑一边,或者可以使用数据库存储吗,希望老师讲解一下

    2023-04-07

  • ༺ღ天口²º²²ღ༻ 👍(1) 💬(1)

    老师,FAQ数据在哪里?

    2023-05-06

  • 孟健 👍(1) 💬(1)

    Meta最近也开源了大语言模型,好像更好一些?

    2023-04-06

  • 地平线 👍(0) 💬(2)

    由于llama-index 升级,我使用的版本是0.6.8,修改了llama-index代码,程序运行没有报错,但是没有输出内容 from langchain.text_splitter import SpacyTextSplitter llm_predictor = LLMPredictor(llm=CustomLLM()) text_splitter = CharacterTextSplitter(separator="\n\n", chunk_size=100, chunk_overlap=20) parser = SimpleNodeParser(text_splitter=text_splitter) documents = SimpleDirectoryReader('./drive/MyDrive/colab_data/faq/').load_data() nodes = parser.get_nodes_from_documents(documents) embed_model = LangchainEmbedding(HuggingFaceEmbeddings( model_name="sentence-transformers/paraphrase-multilingual-mpnet-base-v2" )) service_context = ServiceContext.from_defaults(embed_model=embed_model, llm_predictor=llm_predictor) dimension = 768 faiss_index = faiss.IndexFlatIP(dimension) # index = GPTListIndex(nodes=nodes, faiss_index=faiss_index, service_context=service_context) index = GPTListIndex(nodes=nodes, service_context=service_context) from llama_index import QuestionAnswerPrompt QA_PROMPT_TMPL = ( "{context_str}" "\n\n" "根据以上信息,请回答下面的问题:\n" "Q: {query_str}\n" ) QA_PROMPT = QuestionAnswerPrompt(QA_PROMPT_TMPL) query_engine = index.as_query_engine( retriever_mode="embedding", verbose=True, text_qa_template=QA_PROMPT, ) response = query_engine.query("请问你们海南能发货吗?") print(response)

    2023-05-18

  • 茶桁 👍(0) 💬(1)

    老师,“GPTFaissIndex”这个方法似乎没有了,替换的其他方法不知道是什么,怎么用。

    2023-05-18

  • horn 👍(0) 💬(1)

    Colab报错怎么回事呢 ImportError: cannot import name 'GPTFaissIndex' from 'llama_index' (/usr/local/lib/python3.10/dist-packages/llama_index/__init__.py)

    2023-05-15

  • haha 👍(0) 💬(1)

    PaLM2呢,今天被公众号刷屏了

    2023-05-11

  • Geek_9znsx3 👍(0) 💬(1)

    在colab运行11_colab_chatglm_opensource.ipynb,安装python包时报错,protobuf包版本冲突,请问这个怎么解决?从报错信息看是由于tensorflow-2.12.0依赖protobuf3.20.3,但是icetk-0.0.7依赖 protobuf-3.18.3 Installing collected packages: protobuf, icetk Attempting uninstall: protobuf Found existing installation: protobuf 3.20.3 Uninstalling protobuf-3.20.3: Successfully uninstalled protobuf-3.20.3 ERROR: pip's dependency resolver does not currently take into account all the packages that are installed. This behaviour is the source of the following dependency conflicts. ........ tensorflow 2.12.0 requires protobuf!=4.21.0,!=4.21.1,!=4.21.2,!=4.21.3,!=4.21.4,!=4.21.5,<5.0.0dev,>=3.20.3, but you have protobuf 3.18.3 which is incompatible. tensorflow-datasets 4.9.2 requires protobuf>=3.20, but you have protobuf 3.18.3 which is incompatible. tensorflow-hub 0.13.0 requires protobuf>=3.19.6, but you have protobuf 3.18.3 which is incompatible. tensorflow-metadata 1.13.1 requires protobuf<5,>=3.20.3, but you have protobuf 3.18.3 which is incompatible. Successfully installed icetk-0.0.7 protobuf-3.18.3

    2023-05-10

  • 李文龙 👍(0) 💬(3)

    请问使用 llama_index 0.6.1 版本,使用 Faiss,代码如下,有报错。 import faiss from llama_index.vector_stores import FaissVectorStore from llama_index import GPTVectorStoreIndex, StorageContext dimension = 768 faiss_index = faiss.IndexFlatIP(dimension) storage_context = StorageContext.from_defaults( vector_store=FaissVectorStore(faiss_index) ) print(len(nodes)) index = GPTVectorStoreIndex(nodes, storage_context=storage_context) query_engine = index.as_query_engine( retriever_mode="embedding", verbose=True, service_context = service_context, ) 报错: /usr/local/lib/python3.10/dist-packages/faiss/__init__.py in replacement_add(self, x) 212 213 n, d = x.shape --> 214 assert d == self.d 215 self.add_c(n, swig_ptr(x)) 216

    2023-05-05

  • Santiago 👍(0) 💬(4)

    老师我想问一下 我用colab报错 KeyError: '-1' 这是咋回事呀,还有就是我用3060的显卡6G,直接爆显存了

    2023-04-20

  • Santiago 👍(0) 💬(1)

    RuntimeError: Failed to import transformers.models.t5.configuration_t5 because of the following error (look up to see its traceback): Failed to import transformers.onnx.config because of the following error (look up to see its traceback): DLL load failed while importing _imaging: 找不到指定的模块。

    2023-04-19