Steve Hu的博客

最爱哀酱!

1
2
3
4
5
6
from transformers import AutoTokenizer, AutoModel
tokenizer = AutoTokenizer.from_pretrained("THUDM/chatglm2-6b-int4", trust_remote_code=True)
model = AutoModel.from_pretrained("THUDM/chatglm2-6b-int4", trust_remote_code=True).half().float()
model = model.eval()
response, history = model.chat(tokenizer, "写一首赞美雪的现代诗", history=[])
print(response)

人的大脑运作,可以看作是一个函数系统,每个函数,都对应一个输入,产生一个输出。

阅读全文 »

这是我发布的基于HEXO的2.0版本博客,稍后我会写出教程

0%