change config

README.md

@@ -48,18 +48,23 @@ pip install --editable .
 ```
 
 ```python3
-import argparse
 import torch
 from fengshen.models.DAVAE.DAVAEModel import DAVAEModel
+from transformers import BertTokenizer,T5Tokenizer
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+
+encoder_tokenizer = BertTokenizer.from_pretrained("IDEA-CCNL/Randeng-DAVAE-1.2B-General-Chinese")
+decoder_tokenizer = T5Tokenizer.from_pretrained("IDEA-CCNL/Randeng-DAVAE-1.2B-General-Chinese", eos_token = '<|endoftext|>', pad_token = '<pad>',extra_ids=0)
+decoder_tokenizer.add_special_tokens({'bos_token':'<bos>'})
 vae_model = DAVAEModel.from_pretrained("IDEA-CCNL/Randeng-DAVAE-1.2B-General-Chinese").to(device)
 input_texts = [
     "针对电力系统中的混沌振荡对整个互联电网的危害问题,提出了一种基于非线性光滑函数的滑模控制方法.",
     "超市面积不算大.挺方便附近的居民购买的. 生活用品也比较齐全.价格适用中.",
 ]
-output_texts = vae_model.simulate_batch(input_texts)
+output_texts = vae_model.simulate_batch(encoder_tokenizer,decoder_tokenizer,input_texts)
 print(output_texts)
 
+
 ```
 
 ## 引用 Citation

config.json

@@ -1,7 +1,4 @@
 {
-  "encoder_model_path":  "/encoder/",
-  "decoder_model_path":  "/decoder/",
-  "sentencepiece_model_path":"cog-pretrain.model",
   "latent_size":256,
   "seed": 42, 
   "ratio_increase": 0.5,