feat: 微信协议#
This commit is contained in:
@@ -1,5 +1,10 @@
|
||||
https://cdn.discordapp.com/attachments/1089871481128898582/1101348040876494889/mmexport1682652223225.jpg small Fingerprint lock hanging on wardrobe
|
||||
|
||||
|
||||
|
||||
https://cdn.discordapp.com/attachments/1089871481128898582/1101432414112325704/Screenshot_2023-04-28_at_16.58.55.png Open the bedside table
|
||||
```
|
||||
from transformers import AutoTokenizer, AutoModel
|
||||
tokenizer = AutoTokenizer.from_pretrained("THUDM/chatglm2-6b", trust_remote_code=True)
|
||||
model = AutoModel.from_pretrained("THUDM/chatglm2-6b", trust_remote_code=True, device='cuda')
|
||||
model = model.eval()
|
||||
response, history = model.chat(tokenizer, "你好", history=[])
|
||||
rint(response)
|
||||
response, history = model.chat(tokenizer, "晚上睡不着应该怎么办", history=history)
|
||||
print(response)
|
||||
```
|
||||
|
||||
Reference in New Issue
Block a user