update
Browse files
README.md
CHANGED
@@ -25,12 +25,12 @@ GEB-1.3B在MMLU、C-Eval和CMMLU等常用基准测试中表现优异,超过了
|
|
25 |
```python
|
26 |
from transformers import AutoTokenizer, AutoModel
|
27 |
import torch
|
28 |
-
model = AutoModel.from_pretrained("
|
29 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
30 |
-
|
31 |
-
|
32 |
-
|
33 |
-
|
34 |
```
|
35 |
## 协议
|
36 |
|
|
|
25 |
```python
|
26 |
from transformers import AutoTokenizer, AutoModel
|
27 |
import torch
|
28 |
+
model = AutoModel.from_pretrained("GEB-AGI/geb-1.3b", trust_remote_code=True).bfloat16().cuda()
|
29 |
+
tokenizer = AutoTokenizer.from_pretrained("GEB-AGI/geb-1.3b", trust_remote_code=True)
|
30 |
+
|
31 |
+
query = "你好"
|
32 |
+
response, history = model.chat(tokenizer, query, history=[])
|
33 |
+
print(response)
|
34 |
```
|
35 |
## 协议
|
36 |
|