ZhenYang21 commited on
Commit
1821a63
1 Parent(s): 840e994

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +0 -11
README.md CHANGED
@@ -9,18 +9,7 @@ test set.
9
 
10
 
11
  ## How to use
12
- ```python
13
- from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
14
- tokenizer = AutoTokenizer.from_pretrained("BAAI/glm-10b-chinese", trust_remote_code=True)
15
- model = AutoModelForSeq2SeqLM.from_pretrained("BAAI/glm-10b-chinese", trust_remote_code=True)
16
- model = model.half().cuda()
17
 
18
- inputs = tokenizer("凯旋门位于意大利米兰市古城堡旁。1807年为纪念[MASK]而建,门高25米,顶上矗立两武士青铜古兵车铸像。", return_tensors="pt")
19
- inputs = tokenizer.build_inputs_for_generation(inputs, max_gen_length=512)
20
- inputs = {key: value.cuda() for key, value in inputs.items()}
21
- outputs = model.generate(**inputs, max_length=512, eos_token_id=tokenizer.eop_token_id)
22
- print(tokenizer.decode(outputs[0].tolist()))
23
- ```
24
 
25
  ## Citation
26
  Please cite our paper if you find this code useful for your research:
 
9
 
10
 
11
  ## How to use
 
 
 
 
 
12
 
 
 
 
 
 
 
13
 
14
  ## Citation
15
  Please cite our paper if you find this code useful for your research: