Commit
•
dcd0e32
1
Parent(s):
b55fd41
Update README.md
Browse files
README.md
CHANGED
@@ -8,6 +8,8 @@ pipeline_tag: text-classification
|
|
8 |
|
9 |
加载代码:
|
10 |
```python
|
|
|
|
|
11 |
import torch
|
12 |
import os
|
13 |
from transformers import AutoConfig, AutoModel, AutoTokenizer
|
@@ -23,9 +25,8 @@ for k, v in prefix_state_dict.items():
|
|
23 |
if k.startswith("transformer.prefix_encoder."):
|
24 |
new_prefix_state_dict[k[len("transformer.prefix_encoder."):]] = v
|
25 |
model.transformer.prefix_encoder.load_state_dict(new_prefix_state_dict)
|
26 |
-
|
27 |
model.half().cuda().eval()
|
28 |
|
29 |
-
response, history = model.chat(tokenizer, "
|
30 |
print(response)
|
31 |
```
|
|
|
8 |
|
9 |
加载代码:
|
10 |
```python
|
11 |
+
!git clone https://huggingface.co/Jerome2046/glm-emotion
|
12 |
+
|
13 |
import torch
|
14 |
import os
|
15 |
from transformers import AutoConfig, AutoModel, AutoTokenizer
|
|
|
25 |
if k.startswith("transformer.prefix_encoder."):
|
26 |
new_prefix_state_dict[k[len("transformer.prefix_encoder."):]] = v
|
27 |
model.transformer.prefix_encoder.load_state_dict(new_prefix_state_dict)
|
|
|
28 |
model.half().cuda().eval()
|
29 |
|
30 |
+
response, history = model.chat(tokenizer, "说出'到北京五环25分钟车程,938离我住处只有300米,饮食一条街更是北京饮食街的翻版'此文段表达的情绪类型", history=[])
|
31 |
print(response)
|
32 |
```
|