Update README.md

README.md

@@ -17,8 +17,8 @@ There is no structure of Zhouwenwang-1.3B in [Transformers](https://github.com/h
 
 ### Load model
 ```python
-from model.roformer.modeling_roformer import RoFormerModel    
-from model.roformer.configuration_roformer import RoFormerConfig
+from fengshen import RoFormerModel    
+from fengshen import RoFormerConfig
 from transformers import BertTokenizer
 
 tokenizer = BertTokenizer.from_pretrained("IDEA-CCNL/Zhouwenwang-1.3B")
@@ -31,7 +31,7 @@ model = RoFormerModel.from_pretrained("IDEA-CCNL/Zhouwenwang-1.3B")
 You can use Zhouwenwang-1.3B to continue writing
 
 ```python
-from model.roformer.modeling_roformer import RoFormerModel
+from fengshen import RoFormerModel
 from transformers import AutoTokenizer
 import torch
 import numpy as np