junnyu commited on
Commit
3892cf6
1 Parent(s): aed8280

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -9
README.md CHANGED
@@ -32,18 +32,11 @@ datasets:
32
  - [**预训练日志**](https://wandb.ai/junyu/electra_rotary_small_pretrain?workspace=user-junyu)
33
  - [**GLUE微调日志**](https://wandb.ai/junyu/electra_rotary_glue_100?workspace=user-junyu)
34
 
35
- # 五、安装
36
- ```bash
37
- pip install roformer
38
-
39
- pip install git+https://github.com/JunnYu/RoFormer_pytorch.git
40
- ```
41
 
42
- # 六、 使用
43
  ```python
44
  import torch
45
- from roformer import RoFormerModel
46
- from transformers import ElectraTokenizer
47
  tokenizer = ElectraTokenizer.from_pretrained("junnyu/roformer_small_discriminator")
48
  model = RoFormerModel.from_pretrained("junnyu/roformer_small_discriminator")
49
  inputs = tokenizer("Beijing is the capital of China.", return_tensors="pt")
 
32
  - [**预训练日志**](https://wandb.ai/junyu/electra_rotary_small_pretrain?workspace=user-junyu)
33
  - [**GLUE微调日志**](https://wandb.ai/junyu/electra_rotary_glue_100?workspace=user-junyu)
34
 
 
 
 
 
 
 
35
 
36
+ # 五、 使用
37
  ```python
38
  import torch
39
+ from transformers import ElectraTokenizer,RoFormerModel
 
40
  tokenizer = ElectraTokenizer.from_pretrained("junnyu/roformer_small_discriminator")
41
  model = RoFormerModel.from_pretrained("junnyu/roformer_small_discriminator")
42
  inputs = tokenizer("Beijing is the capital of China.", return_tensors="pt")