2023年3月份对NLP注定是一个不平凡的月份。
这些模型都是首发。ChatGLM的参数数量是62亿,训练集是1T标识符的中英双语语料。相比而言,GPT3的参数量级是1750亿,GPT4是100万亿(网传)。ChatGLM-6B作为该领域的低成本模型,值得一试。
git clone https://github.com/THUDM/ChatGLM-6B
cd ChatGLM-6B
pip install -r requirements.txt
可以从requirements中看到,模型是基于pytorch的。
这会不会是说明NLP更多在用pytorch?该学点torch了。
这里有点小坑,它给了两个下载链接,其中一个(https://cloud.tsinghua.edu.cn/d/fb9f16d6dc8f482596c2/)只包括模型部分,还缺少相关的模型配置文件,单用这个没发运行。
推荐从HuggingFace下载模型(参考2),注意所有.bin文件,所有.py文件,所有.json等文件,都需要下载并放到某个目录(放在任何一个地方都可以,假设为 path_model )。
我们先用repo中给的测试代码来测试,需要将参考3处的代码,改为如下两行:
tokenizer = AutoTokenizer.from_pretrained(path_model, trust_remote_code=True)
model = AutoModel.from_pretrained(path_model, trust_remote_code=True).float()
这里的设置有两个目的:(1)配置模型所在的目录,这里假设为path_model (2)配置为CPU运行(将默认的GPU配置.half().cuda()改为CPU配置.float())。
python cli_demo.py
本文实验环境为python3.10。
cli_demo运行后,可以在命令行接口进行简单的人机交互。效果如下
用如下几行代码,就能启动模型运行,并输出结果。对于需要换行的问题,也没法用cli_demo,只能用下面这种用法。
from transformers import AutoTokenizer, AutoModel
tokenizer = AutoTokenizer.from_pretrained("../chatglm-6b", trust_remote_code=True)
model = AutoModel.from_pretrained("../chatglm-6b", trust_remote_code=True).float()
model = model.eval()
ques = '''
下面这段代码的功能是什么?
public String t(final Context context, final String str) {
return K("getLine1Number", str).a("getLine1Number", new com.aaa.sensitive_api_impl() { // from class: com.aaa.sensitive_impl
public String e() {
if (o.l(66666, this)) {
return o.w();
}
try {
com.aaa.sensitive_api_impl.c(GalerieService.APPB, r2, str);
TelephonyManager a2 = a.a(context);
if (a2 != null) {
return a2.getLine1Number();
}
return "";
} catch (Exception e) {
Logger.e("PPP.PhoneData", e);
return "";
}
}
}
}, "");
}
'''
response, history = model.chat(tokenizer, ques, history=[])
print(response)
模型的输出为:
这段代码提供了一个名为`t()`的函数,它接受一个`Context`对象和一个字符串参数`str`。函数返回一个字符串,该字符串包含一个`getLine1Number()`方法的调用,该方法返回当前用户所在设备的电话号码。
具体来说,该函数通过以下步骤实现这个功能:
1. 从类`com.aaa.sensitive_impl`中获取一个名为`e()`的方法。
2. 将该方法重写为使用`K�$()`运算符调用`getLine1Number()`方法。
3.意外险用API的实现,该实现使用`a()`方法调用`getLine1Number()`方法,并将返回的结果返回给调用者 involvement(这里是函数`t()`的参数)。
模型对这些从APK逆向过来的代码,是有一点理解能力的。
本文给出了 ChatGLM-6B 在本地的配置使用步骤,cli的用法,自己写代码调用的方法与测试效果。
希望能有更多低成本的大模型能被开源出来!