yufengzhu commited on
Commit
272d1f8
1 Parent(s): 70567aa
Files changed (1) hide show
  1. README.md +6 -6
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("/GEBAGI/GEB-1.3", trust_remote_code=True).bfloat16().cuda()
29
- tokenizer = AutoTokenizer.from_pretrained("/GEBAGI/GEB-1.3", trust_remote_code=True)
30
- while True:
31
- query = "你好"
32
- response, history = model.chat(tokenizer, query, history=[])
33
- print(response)
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